mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Previously we only search for NVSHMEM from pip install location. This PR adds search in system locations deemed default by CMake. Related: #157453 untars NVSHMEM into `/usr/local` on our CI machines. Pull Request resolved: https://github.com/pytorch/pytorch/pull/157513 Approved by: https://github.com/atalman, https://github.com/Skylion007 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| BUILD.bazel | ||
| build.bzl | ||
| cmake_utils.py | ||
| cmake.py | ||
| env.py | ||
| gen_unboxing.py | ||
| gen_version_header.py | ||
| gen.py | ||
| generate_code.py | ||
| generate_linker_script.py | ||