From 1a2b41dbbcec71214b1abed9ed77d11d57701aef Mon Sep 17 00:00:00 2001 From: liqun Fu Date: Mon, 9 Aug 2021 16:49:59 -0700 Subject: [PATCH] packaging pipeline produces -cpu- named packages due to a logical error (#8665) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 1d6246f63f..e1b265b860 100644 --- a/setup.py +++ b/setup.py @@ -327,7 +327,7 @@ if enable_training: + 'cu' + cuda_version.replace('.', '') else: local_version = '+cu' + cuda_version.replace('.', '') - if rocm_version: + elif rocm_version: # removing '.' to make Cuda version number in the same form as Pytorch. rocm_version = rocm_version.replace('.', '') if torch_version: