Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Select an option

  • Save davidberard98/0304f27ecb8559b44402a0b9255be055 to your computer and use it in GitHub Desktop.

Select an option

Save davidberard98/0304f27ecb8559b44402a0b9255be055 to your computer and use it in GitHub Desktop.
TORCHINDUCTOR_FORCE_DISABLE_CACHES=1
function finish() {
pushd /home/dberard/local/pytorch-env7/triton
git checkout -- scripts/build-llvm-project.sh
}
trap finish EXIT
git patch /home/dberard/local/pytorch-env7/diff.patch
make dev-install-llvm
code=$?
if [ $code -ne 0 ]
then
exit 125
fi
pushd ../pytorch
# python test/inductor/test_torchinductor.py -k triton_kernel
# python test/inductor/test_cooperative_reductions.py -k test_welford_non_power_of_2_rsplit_persistent_True_x_9_r_8000_rsplit_37
rm /home/dberard/local/miniconda3/envs/pytorch-env7/lib/python3.12/site-packages/segment_anything_fast/configs/flash_4_configs_a100.p
rm /home/dberard/local/pytorch-env7/pytorch/flash_4_configs.p
rm -rf /tmp/torchinductor_dberard
python benchmarks/dynamo/torchbench.py --accuracy --timing --explain --print-compilation-time --inductor --device cuda --inference --bfloat16 --only sam_fast 2>&1 | tee torchbench.log
cat torchbench.log | grep eager_fail_to_run
code=$?
if [ $code -ne 0 ]
then
echo "did NOT find eager_fail_to_run, so exit 0"
exit 0
fi
echo "found eager_fail_to_run, so exit 1"
exit 1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment