[FEA]: Add support for pathfinder.find_bitcode_lib("nvshmem_device")#1828
[FEA]: Add support for pathfinder.find_bitcode_lib("nvshmem_device")#1828rwgk wants to merge 2 commits intoNVIDIA:mainfrom
pathfinder.find_bitcode_lib("nvshmem_device")#1828Conversation
(site-packages, Conda) x (cu12, cu13)
rwgk
added
P1
|
Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually. Contributors can view more details about this message here. |
nvshmem wheels are not published for Windows, so the nvshmem_device entry must be filtered out on that platform. This adds an `available_on_windows` key to _BitcodeLibInfo and uses it to build SUPPORTED_BITCODE_LIBS, which controls test parametrization. Made-with: Cursor
|
/ok to test |
|
Hi @rwgk - thanks for this! Right now, there is in fact only one libnvshmem_device.bc. In NVSHMEM 3.7, there will probably be a libnvshmem_device.bc, and a libnvshmem_device.ltoir. There may also be Thanks |
|
Conclusion from offline discussion with @benhg: the current state of the PR is good to go. In the future we can add e.g. |
2 participants
Closes #1421
Tested with (site-packages, Conda) x (cu12, cu13) on Linux.