From b7129bd81020d8d11e3d58266bc010c7792e908e Mon Sep 17 00:00:00 2001 From: hjjq Date: Mon, 18 Dec 2023 19:56:16 -0500 Subject: [PATCH] Fix reduce bug according to #372 --- python/hidet/graph/ops/reduce/reduce.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/hidet/graph/ops/reduce/reduce.py b/python/hidet/graph/ops/reduce/reduce.py index 4aecff190..7a0979aac 100644 --- 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