Skip to content

GitLab

Unverified Commit 5a3e6ebf authored by Sage Moore's avatar Sage Moore Committed by GitHub
Browse files

Update torch to 2.4.0 (#8) 


Co-authored-by: default avatarWoosuk Kwon <woosuk.kwon@berkeley.edu>
parent 8f48a546
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
.github/workflows/publish.yml
View file @ 5a3e6ebf
...@@ -49,7 +49,7 @@ jobs: ...@@ -49,7 +49,7 @@ jobs:
matrix: matrix:
os: ['ubuntu-20.04'] os: ['ubuntu-20.04']
python-version: ['3.8', '3.9', '3.10', '3.11'] python-version: ['3.8', '3.9', '3.10', '3.11']
pytorch-version: ['2.3.1'] # Should be synced with setup.py. pytorch-version: ['2.4.0'] # Should be synced with setup.py.
cuda-version: ['12.1'] cuda-version: ['12.1']
steps: steps:
......
build.sh
View file @ 5a3e6ebf
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
# A simple build script for local testing. # A simple build script for local testing.
# NOTE: This script is not used for the actual build process. # NOTE: This script is not used for the actual build process.
PYTORCH_VERSION="2.3.1" PYTORCH_VERSION="2.4.0"
pip install packaging ninja; pip install packaging ninja;
pip install torch==${PYTORCH_VERSION}; pip install torch==${PYTORCH_VERSION};
......
setup.py
View file @ 5a3e6ebf
...@@ -263,7 +263,7 @@ class NinjaBuildExtension(BuildExtension): ...@@ -263,7 +263,7 @@ class NinjaBuildExtension(BuildExtension):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
PYTORCH_VERSION = "2.3.1" PYTORCH_VERSION = "2.4.0"
CUDA_VERSION = "12.1" CUDA_VERSION = "12.1"
setup( setup(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment