We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 031bb11 commit c01d1baCopy full SHA for c01d1ba
torchao/prototype/inductor/fx_passes/int8_sdpa_fusion.py
@@ -18,7 +18,7 @@
18
from torchao.utils import TORCH_VERSION_AT_LEAST_2_7
19
20
if TORCH_VERSION_AT_LEAST_2_7:
21
- # TORCH_VERSION_AT_LEAST_2_7 is needed for int8 sdpa lowering
+ # TORCH_VERSION_AT_LEAST_2_7 is needed for functions in int8 sdpa lowering
22
from ..int8_sdpa_lowering import register_int8_sdpa # noqa: F401
23
else:
24
make_fallback(torch.ops.torchao.qscaled_dot_product.default)
0 commit comments