Skip to content

Commit

Permalink
[CI] Switch inference accuracy and performance tests to bfloat16
Browse files Browse the repository at this point in the history
ghstack-source-id: 4ccf078b2ed80187f3fa6b00f16181195fe70425
Pull Request resolved: #103535
  • Loading branch information
desertfire committed Jun 13, 2023
1 parent 51d21ff commit e536a96
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion .ci/pytorch/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -319,6 +319,7 @@ test_perf_for_dashboard() {
fi
if [[ "$DASHBOARD_TAG" == *inference-true* ]]; then
modes+=(inference)
dtype=bfloat16
fi
# TODO: All the accuracy tests can be skipped once the CI accuracy checking is stable enough
local targets=(accuracy performance)
Expand Down Expand Up @@ -429,7 +430,7 @@ test_dynamo_benchmark() {
if [[ "${TEST_CONFIG}" == *cpu_accuracy* ]]; then
test_single_dynamo_benchmark "inference" "$suite" "$shard_id" --inference --float32 "$@"
else
test_single_dynamo_benchmark "inference" "$suite" "$shard_id" --inference --amp "$@"
test_single_dynamo_benchmark "inference" "$suite" "$shard_id" --inference --bfloat16 "$@"
test_single_dynamo_benchmark "training" "$suite" "$shard_id" --training --amp "$@"
fi
fi
Expand Down

0 comments on commit e536a96

Please sign in to comment.