Coverage for src/flag_gems/runtime/backend/_nvidia/hopper/ops/__init__.py: 0%

1import triton 

2 

3if triton.__version__ >= "3.4": 

4 from .fill import ( # noqa: F401 

5 fill_scalar, 

6 fill_scalar_, 

7 fill_scalar_out, 

8 fill_tensor, 

9 fill_tensor_, 

10 fill_tensor_out, 

11 ) 

12 from .mm import mm, mm_out, router_gemm # noqa: F401 

13 from .mul import mul # noqa: F401 

14 from .sqrt import sqrt, sqrt_ # noqa: F401 

15 from .w8a8_block_fp8_matmul import w8a8_block_fp8_matmul # noqa: F401 

16 

17# The Gluon FP8 block-wise BMM kernel and fp8_einsum require Triton >= 3.6.0. 

18if triton.__version__ >= "3.6.0": 

19 from .fp8_einsum import fp8_einsum # noqa: F401 

20 from .w8a8_block_fp8_bmm import w8a8_block_fp8_bmm # noqa: F401 

21 

22__all__ = ["*"]