search

pytorch / ao

PyTorch native quantization and sparsity for training and inference
BSD 3-Clause "New" or "Revised" License
1.6k stars 179 forks source link

Skipping nvcc compilation for sm equal or lower than 80 #1323

Closed jerryzh168 closed 4 days ago

jerryzh168 commented 4 days ago

sm_80 or lower does not have bfloat16 which causes the compilation to fail, e.g. https://github.com/pytorch/pytorch/actions/runs/11960091588/job/33347870789

pytorch-bot[bot] commented 4 days ago

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1323

Note: Links to docs will display an error until the docs builds have been completed.

:heavy_exclamation_mark: 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

:white_check_mark: No Failures

As of commit 204d0c781153eac7030dc7f8b588a864be7b3cae with merge base 7489c7d08a07f5003cf393f1cdf56c170af47e1c (image): :green_heart: Looks good so far! There are no failures yet. :green_heart:

This comment was automatically generated by Dr. CI and updates every 15 minutes.