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 cfec2ef commit e4b3c06Copy full SHA for e4b3c06
tilelang/jit/adapter/tvm_ffi.py
@@ -190,7 +190,7 @@ def torch_dtype_to_tvm_str(dtype: torch.dtype) -> str:
190
return "float8_e5m2"
191
# Strip torch. prefix for readability
192
s = str(dtype)
193
- return s[6:] if s.startswith("torch.") else s
+ return s[6:] if s.startswith("paddle.") else s
194
195
def func(*inputs: torch.Tensor | Any):
196
# Validate input count strictly
0 commit comments