diff --git a/tests/unit/test_int32_cast.py b/tests/unit/test_int32_cast.py index 5521b054..5d1daa11 100644 --- a/tests/unit/test_int32_cast.py +++ b/tests/unit/test_int32_cast.py @@ -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)