NVFP4-1.1: try .to(Int32) for float-to-int conversion

This commit is contained in:
2026-05-28 04:02:45 +00:00
parent dc35d29811
commit c3d5a7b82f

View File

@@ -13,7 +13,7 @@ def test_int32_cast_kernel(
tidx, _, _ = cute.arch.thread_idx()
if tidx == cutlass.Int32(0):
f = cute.arch.load(input_f32.iterator, cutlass.Float32)
i = cutlass.Int32(f) # float-to-int conversion
i = f.to(cutlass.Int32) # float-to-int conversion
cute.arch.store(output_i32.iterator, i)