Compare commits

...

1 Commits

Author SHA1 Message Date
e35df4c13b [Experiment only] Collect cpp_wrapper perf numbers 2024-12-02 14:43:21 -05:00

View File

@ -487,6 +487,8 @@ test_perf_for_dashboard() {
device=cuda_a10g
fi
export TORCHINDUCTOR_CPP_WRAPPER=1
for mode in "${modes[@]}"; do
if [[ "$mode" == "inference" ]]; then
dtype=bfloat16