NVFP4-1.1: try .to(Int32) for float-to-int conversion
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user