Skip to content

Commit 5b942b3

Browse files
committed
Merge branch 'v0.5.0-release' of https://github.com/pytorch/FBGEMM into v0.5.0-release
2 parents 3e6bb76 + 10d9ec3 commit 5b942b3

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

.github/scripts/fbgemm_gpu_build.bash

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ __configure_fbgemm_gpu_build_cuda () {
104104
else
105105
echo "[BUILD] Using the default CUDA targets ..."
106106
# For cuda version 12.1, enable sm 9.0
107-
cuda_version_nvcc=$(nvcc --version)
107+
cuda_version_nvcc=$(conda run -n "${env_name}" nvcc --version)
108108
echo "$cuda_version_nvcc"
109109
if [[ $cuda_version_nvcc == *"V12.1"* ]]; then
110110
local arch_list="7.0;8.0;9.0"

fbgemm_gpu/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def generate_package_version(package_name: str):
4848
print("version from gitversion: ", version)
4949
print(gitversion.version_from_git())
5050
# Manually change version
51-
version = "0.5.0rc1"
51+
version = "0.5.0rc2"
5252

5353
print(f"[SETUP.PY] Setting the package version: {version}")
5454
return version

0 commit comments

Comments
 (0)