singlaamitesh opened a new pull request, #20033:
URL: https://github.com/apache/tvm/pull/20033
Fixes #20031
## Summary
Cold self-review found the libinfo.py fix duplicated an existing mechanism
(tvm.contrib.cutlass.has_cutlass(), which already probes the
'relax.ext.cutlass' global function registered only under USE_CUTLASS=ON) using
a different, less-canonical symbol, and was missing the codebase's standard
import-outside-toplevel pylint-disable on a local import. Aligned libinfo() to
check the same 'relax.ext.cutlass' symbol as has_cutlass(), and updated both
regression tests to call has_cutlass() directly instead of re-deriving the
check. The GEMM ${A_arg}->${lhs_arg} template fix and the missing
tvm/ffi/container/tensor.h header fix were re-verified unchanged and correct.
## Changes
- `python/tvm/support/libinfo.py`
- `python/tvm/contrib/cutlass/gemm_operation.py`
- `python/tvm/contrib/cutlass/gen_tensor_op.py`
- `tests/python/support/test_libinfo.py`
- `tests/python/testing/test_env.py`
- `tests/python/contrib/test_cutlass_gemm.py`
## How this was tested
Ran `tests/` locally.
--
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]
