frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Prepare for Azure Pipeline for multi-gpu tests (#55600)

Browse Source 
Summary:
Previous PR: https://github.com/pytorch/pytorch/issues/52490

Pull Request resolved: https://github.com/pytorch/pytorch/pull/55600

Reviewed By: albanD, seemethere

Differential Revision: D27667544

Pulled By: malfet

fbshipit-source-id: f5843379807d8c95f3791d19ac0ab2d1973fa087
This commit is contained in:
Chester Liu
2021-04-14 09:59:06 -07:00
committed by Facebook GitHub Bot
parent 1127bab828
commit ba320cec6b
4 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.circleci/config.yml
View File

@ -350,7 +350,7 @@ pytorch_windows_params: &pytorch_windows_params
VC_YEAR: <<parameters.vc_year>>
VC_PRODUCT: <<parameters.vc_product>>
USE_CUDA: <<parameters.use_cuda>>
TORCH_CUDA_ARCH_LIST: "7.5"
TORCH_CUDA_ARCH_LIST: "5.2 7.5"
JOB_BASE_NAME: <<parameters.test_name>>
JOB_EXECUTOR: <<parameters.executor>>
binary_linux_build_params: &binary_linux_build_params

5
.circleci/regenerate.ps1 Normal file
View File

@ -0,0 +1,5 @@
cd $PSScriptRoot;
$NewFile = New-TemporaryFile;
python generate_config_yml.py > $NewFile.name
(Get-Content $NewFile.name -Raw).TrimEnd().Replace("`r`n","`n") | Set-Content config.yml -Force
Remove-Item $NewFile.name

2
.circleci/verbatim-sources/build-parameters/pytorch-build-params.yml
View File

@ -88,6 +88,6 @@ pytorch_windows_params: &pytorch_windows_params
VC_YEAR: <<parameters.vc_year>>
VC_PRODUCT: <<parameters.vc_product>>
USE_CUDA: <<parameters.use_cuda>>
TORCH_CUDA_ARCH_LIST: "7.5"
TORCH_CUDA_ARCH_LIST: "5.2 7.5"
JOB_BASE_NAME: <<parameters.test_name>>
JOB_EXECUTOR: <<parameters.executor>>

12
.jenkins/pytorch/win-test-helpers/test_distributed.bat Normal file
View File

@ -0,0 +1,12 @@
REM The first argument should lead to the python interpreter
%1\python.exe test/run_test.py --verbose -i distributed/test_jit_c10d
if %errorlevel% neq 0 ( exit /b %errorlevel% )
%1\python.exe test/run_test.py --verbose -i distributed/test_c10d
if %errorlevel% neq 0 ( exit /b %errorlevel% )
%1\python test/run_test.py --verbose -i distributed/test_c10d_spawn
if %errorlevel% neq 0 ( exit /b %errorlevel% )
%1\python.exe test/run_test.py --verbose -i distributed/test_data_parallel
if %errorlevel% neq 0 ( exit /b %errorlevel% )