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 a4a7716 commit 4293a60Copy full SHA for 4293a60
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