pytorch
1407e672 - Nvfuser python api patch take 2 (#83684)

Commit
3 years ago
Nvfuser python api patch take 2 (#83684) landing #83645 again. Previously we are breaking on codegen bf16 kernel for cuda TK 10.2. Added a short-cut to disable bf tests on pre cuda 11 build. Pull Request resolved: https://github.com/pytorch/pytorch/pull/83684 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading