Convert (m, n, k) to integer

EikanWang · EikanWang · commit 0a0aedd2b6d8 · 2025-10-11T05:30:04.000Z
diff --git a/helion/_compat.py b/helion/_compat.py
@@ -79,9 +79,10 @@ def _min_dot_size(
             if not k.startswith("_")
         }
 
-        return min_dot_size_xpu(gpu_target_info)(
+        dot_size_val = min_dot_size_xpu(gpu_target_info)(
             torch_dtype_to_tl(lhs), torch_dtype_to_tl(rhs)
         )
+        return tuple(int(v) for v in dot_size_val)  # pyright: ignore[reportReturnType]
 
     from triton.backends.nvidia.compiler import min_dot_size as min_dot_size_cuda
 

Original file line number	Diff line number	Diff line change
`@@ -79,9 +79,10 @@ def _min_dot_size(`
`79`	`79`	`if not k.startswith("_")`
`80`	`80`	`}`
`81`	`81`
`82`		`- return min_dot_size_xpu(gpu_target_info)(`
	`82`	`+ dot_size_val = min_dot_size_xpu(gpu_target_info)(`
`83`	`83`	`torch_dtype_to_tl(lhs), torch_dtype_to_tl(rhs)`
`84`	`84`	`)`
	`85`	`+ return tuple(int(v) for v in dot_size_val) # pyright: ignore[reportReturnType]`
`85`	`86`
`86`	`87`	`from triton.backends.nvidia.compiler import min_dot_size as min_dot_size_cuda`
`87`	`88`