Skip to content

Commit e4b3c06

Browse files
committed
update dtype literals
1 parent cfec2ef commit e4b3c06

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tilelang/jit/adapter/tvm_ffi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def torch_dtype_to_tvm_str(dtype: torch.dtype) -> str:
190190
return "float8_e5m2"
191191
# Strip torch. prefix for readability
192192
s = str(dtype)
193-
return s[6:] if s.startswith("torch.") else s
193+
return s[6:] if s.startswith("paddle.") else s
194194

195195
def func(*inputs: torch.Tensor | Any):
196196
# Validate input count strictly

0 commit comments

Comments
 (0)