From f36a093c20963a3827be0cf4fd5e15a0cac69f31 Mon Sep 17 00:00:00 2001 From: anonymousdouble <112695649+anonymousdouble@users.noreply.github.com> Date: Wed, 20 Dec 2023 20:09:30 +1100 Subject: [PATCH] Update conv2d.py (#16262) refactor with chain constant value assignment to make code more Pythonic, concise and efficient. --- python/tvm/topi/intel_graphics/conv2d.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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")