This is an automated email from the ASF dual-hosted git repository.

sanirudh pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new f36a093c20 Update conv2d.py (#16262)
f36a093c20 is described below

commit f36a093c20963a3827be0cf4fd5e15a0cac69f31
Author: anonymousdouble <[email protected]>
AuthorDate: Wed Dec 20 20:09:30 2023 +1100

    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 b276bcae92..b7906cdb91 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")

Reply via email to