Skip to content

Commit 634849d

Browse files
allenwang28facebook-github-bot
authored andcommitted
Build with nightly Torch instead of the latest
Summary: Changes in the PyTorch C++ layer affect Monarch's tensor engine builds. Without this change, the CI builds will pick up the latest PyTorch through build-requirements.txt. This diff removes this dependency. We see this affecting #1680 Differential Revision: D85670693
1 parent 9b48962 commit 634849d

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

.github/workflows/build-cuda.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ jobs:
3333
3434
# Setup build environment (conda + system deps + rust + build deps)
3535
setup_build_environment
36+
setup_pytorch_with_headers "${{ matrix.gpu-arch-version }}" "${{ matrix.torch-spec }}"
3637
3738
# Setup Tensor Engine
3839
setup_tensor_engine

build-requirements.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
torch
1+
# torch is also a dependency here, but we want to ensure that
2+
# we're building against the PyTorch index for nightly.
23
setuptools
34
setuptools-rust
45
wheel

0 commit comments

Comments
 (0)