diff --git a/torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.py b/torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.py index 96fe53e92065f3..a9212e088324be 100644 --- a/torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.py +++ b/torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.py @@ -53,18 +53,11 @@ TileDescription, ) - import torch from torch._inductor.codegen.cuda import cuda_env from torch._inductor.utils import IndentedBuffer _CUTLASS_C_DTYPES = OrderedSet(dtype2ctype.values()) # type: ignore[var-annotated] - TORCH_TO_CUTLASS_DTYPE = { - torch.float32: DataType.f32, - torch.float16: DataType.f16, - torch.bfloat16: DataType.bf16, - } - def create_example_tensors( read_names: list[str], write_names: list[str],