Fix reduce bug according to #372

hidet-org · Dec 19, 2023 · b7129bd · b7129bd
1 parent 340906c
commit b7129bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/hidet/graph/ops/reduce/reduce.py b/python/hidet/graph/ops/reduce/reduce.py
@@ -222,7 +222,7 @@ def cuda_schedule_reduce_by_default(self, max_block_size=256, use_atomic=True) -
             num_eles: int = 4 // xdtype.nbytes
             if shape[-1] % num_eles == 0:
                 lanes = num_eles
-                vtype = VectorType(xdtype, lanes)
+                vtype = vectorize(xdtype, lanes)
 
         read_shape = shape[:]
         read_shape[-1] //= lanes