Nanmur opened a new pull request, #19970: URL: https://github.com/apache/tvm/pull/19970
This PR fixes an undefined `TX` reference in the DLight GPU GEMV inner-reduction schedule. The broadcast epilogue path splits fused epilogue loops and binds the inner loop to `threadIdx.x`, but it referenced `TX`, which is not defined in the surrounding scope or passed into the helper. This PR uses the existing `TR` tile factor, which is already used for the `threadIdx.x` direction in this schedule. A regression test is added to cover the GEMV broadcast epilogue path. Fixes #19969 Tests: - `python -m pytest tests/python/s_tir/dlight/test_gpu_gemv.py -q` - `python -m pytest tests/python/s_tir/dlight/test_gpu_low_batch_gemv.py -q` - `python -m pytest tests/python/s_tir/dlight/test_gpu_reduction.py -q` Note: The branch is based on `apache/tvm:main`. Local main-branch test execution on this Windows machine could not be completed because the available compiled TVM library is from a v0.25 build and does not match the latest Python sources; the same tests passed in the matching local v0.25 environment before rebasing the patch to main. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
