8000 [Cutlass] Remove unused dtype conversion map · pytorch/pytorch@455ab41 · GitHub
[go: up one dir, main page]

Skip to content

Commit 455ab41

Browse files
committed
[Cutlass] Remove unused dtype conversion map
ghstack-source-id: 893fce4 Pull Request resolved: #152305
1 parent 99b6c42 commit 455ab41

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -53,18 +53,11 @@
5353
TileDescription,
5454
)
5555

56-
import torch
5756
from torch._inductor.codegen.cuda import cuda_env
5857
from torch._inductor.utils import IndentedBuffer
5958

6059
_CUTLASS_C_DTYPES = OrderedSet(dtype2ctype.values()) # type: ignore[var-annotated]
6160

62-
TORCH_TO_CUTLASS_DTYPE = {
63-
torch.float32: DataType.f32,
64-
torch.float16: DataType.f16,
65-
torch.bfloat16: DataType.bf16,
66-
}
67-
6861
def create_example_tensors(
6962
read_names: list[str],
7063
write_names: list[str],

0 commit comments

Comments
 (0)
0