8000
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 99b6c42 commit 455ab41Copy full SHA for 455ab41
torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.py
@@ -53,18 +53,11 @@
53
TileDescription,
54
)
55
56
- import torch
57
from torch._inductor.codegen.cuda import cuda_env
58
from torch._inductor.utils import IndentedBuffer
59
60
_CUTLASS_C_DTYPES = OrderedSet(dtype2ctype.values()) # type: ignore[var-annotated]
61
62
- TORCH_TO_CUTLASS_DTYPE = {
63
- torch.float32: DataType.f32,
64
- torch.float16: DataType.f16,
65
- torch.bfloat16: DataType.bf16,
66
- }
67
-
68
def create_example_tensors(
69
read_names: list[str],
70
write_names: list[str],
0 commit comments