Fix incorrect example paths in CuTeDSL docstrings#3151
Open
Weili-0234 wants to merge 2 commits into
Open
Conversation
The CuTeDSL examples reference incorrect paths in their docstrings and usage comments. All paths were missing the `python/CuTeDSL/` prefix (e.g., `examples/ampere/sgemm.py` instead of `examples/python/CuTeDSL/ampere/sgemm.py`). Some files under `cute/tvm_ffi/` also used a stale `cutlass_ir/compiler/python/` prefix. This is a superset of the fix in PR NVIDIA#2741, covering all 69 affected files across ampere, hopper, blackwell, cute, distributed, jax, and experimental subdirectories.
Addresses issues missed by the batch path prefix replacement: - cooperative_launch.py, dynamic_smem_size.py: add full path prefix - dense_blockscaled_gemm_persistent_amax.py: fix missing _amax suffix - mla_decode_fp16.py, mla_decode_fp8.py: fix filename and add mla/ subdir - grouped_mixed_input_gemm.py: add missing mixed_input_gemm/ subdir - grouped_gemm.py (hopper): add full path prefix - distributed_gemm_all_reduce_blackwell.py: fix wrong filename - elementwise_apply_example.py (jax): fix references to wrong directory
|
This PR has been labeled |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This PR fix incorrect file paths in CuTeDSL example docstrings that prevent users from copy-pasting run commands directly. This is a superset of #2741, covering all affected files.
Three categories of issues fixed:
python/CuTeDSL/prefix (69 files) e.g.examples/ampere/sgemm.py→examples/python/CuTeDSL/ampere/sgemm.pycutlass_ir/compiler/python/prefix (6 files undercute/tvm_ffi/) e.g.cutlass_ir/compiler/python/examples/cute/tvm_ffi/aot_export.py→examples/python/CuTeDSL/cute/tvm_ffi/aot_export.pyblackwell/mla_fp16.py→blackwell/mla/mla_decode_fp16.py