Update conv2d.py (#16262)

refactor with chain constant value assignment to make code more Pythonic, concise and efficient.
apache · Dec 20, 2023 · f36a093 · f36a093
1 parent 3a57a40
commit f36a093
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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")