Skip to content

Commit 611bf3b

Browse files
HGbodaai-mrkogao
andauthored
Fix: Change variable i to x in split operation in cross_compilation_and_rpc.py (#17743)
Co-authored-by: David Ko <[email protected]>
1 parent a7895a3 commit 611bf3b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docs/how_to/tutorials/cross_compilation_and_rpc.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -234,9 +234,9 @@ def run_opencl():
234234
mod = tvm.IRModule.from_expr(te.create_prim_func([A, B]))
235235
sch = tvm.tir.Schedule(mod)
236236
(x,) = sch.get_loops(block=sch.get_block("B"))
237-
xo, xi = sch.split(i, [None, 32])
238-
sch.bind(x, "blockIdx.x")
239-
sch.bind(x, "threadIdx.x")
237+
xo, xi = sch.split(x, [None, 32])
238+
sch.bind(xo, "blockIdx.x")
239+
sch.bind(xi, "threadIdx.x")
240240
func = tvm.compile(sch.mod, target=target)
241241

242242
remote = rpc.connect(opencl_device_host, opencl_device_port)

0 commit comments

Comments
 (0)