diff --git a/python/tvm/topi/intel_graphics/conv2d.py b/python/tvm/topi/intel_graphics/conv2d.py index b276bcae92b1..b7906cdb9108 100644 --- a/python/tvm/topi/intel_graphics/conv2d.py +++ b/python/tvm/topi/intel_graphics/conv2d.py @@ -591,8 +591,7 @@ def _schedule_cl_spatialpack(s, op): OUTPUT_BLOCK_WIDTH = attrs["block_w"] # schedule conv - z_factor = 1 - y_factor = 1 + y_factor = z_factor = 1 x_factor = 16 thread_z = te.thread_axis((0, z_factor), "threadIdx.z") thread_y = te.thread_axis((0, y_factor), "threadIdx.y")