tree cf3a54b138390c15be881c27daee318017e49dda
parent 38861ba39fbdbf44439ea127f66851df5517817d
author Fuzzkatt <zonghan2000@gmail.com> 1690406724 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1690406724 +0000

add skip if sm < 80 check (#105888)

Fix issue where we were testing `test_schema_correctness_nn_functional_scaled_dot_product_attention_cuda_bfloat16` from `test_schema_check.py` on V100, but bfloat16 support on cuda doesn't exist for sm < 80. Added skip if sm < 80 to the failing test. cc @ptrblck @eqy
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105888
Approved by: https://github.com/kit1980