Open
Conversation
superleo
commented
Mar 4, 2026
- docs/develop/musa_gpu_support.md: design, architecture, build, testing
- 3rdparty/gpus/test_cuda_regression.sh: verify CUDA not regressed
- 3rdparty/gpus/test_musa_build.sh: validate MUSA build layout
- 3rdparty/gpus/test_musa_python_compat.py: Python device abstraction
5adcf99 to
69fcd76
Compare
wangxiyu191
reviewed
Mar 6, 2026
Collaborator
wangxiyu191
left a comment
There was a problem hiding this comment.
通过抽象出_device_mod和GpuStream等来实现兼容不同GPU运行环境的方式很合理。可以按照这个思路来实现MUSA兼容。
superleo
added a commit
to superleo/tair-kvcache
that referenced
this pull request
Mar 9, 2026
superleo
added a commit
to superleo/tair-kvcache
that referenced
this pull request
Mar 9, 2026
ea6684e to
c622e0c
Compare
- docs/develop/musa_gpu_support.md: design, architecture, build, testing - 3rdparty/gpus/test_cuda_regression.sh: verify CUDA not regressed - 3rdparty/gpus/test_musa_build.sh: validate MUSA build layout - 3rdparty/gpus/test_musa_python_compat.py: Python device abstraction
…device-agnostic connectors
- .bazelrc, WORKSPACE, 3rdparty/gpus/musa/: add MUSA toolchain and bazel config
(build:musa, build:client_with_musa, musa_configure repository rule)
- client/sdk: MUSA SDK utils (musa_util.h, hf3fs_musa_util.h), buffer check
(sdk_buffer_check_util_musa.{mu,cc}), and genrule for mcc compilation
- client/sdk: MUSA branches in local_file_sdk, transfer_client_impl, hf3fs_sdk,
hf3fs_usrbio_client for stream/memcpy/host-register
- py_connector: device-agnostic GPU module (_get_device_module for cuda/musa),
conditional tl.float8e4nv fallback in batch_gather_scatter_helper
c622e0c to
1971037
Compare
…ns to the sdk_buffer_check_util target.
Author
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
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.



