Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add thrust_11.3 explicitly, remove bad quotes, parallelize nvcc #755

Merged
merged 4 commits into from
May 24, 2021

Conversation

janeyx99
Copy link
Contributor

@janeyx99 janeyx99 commented May 4, 2021

thrust_11.3 was moved into its separate package so we need to explicitly put it as an argument.

The quotes break installation, so removing them as they're not actually necessary

Also parallelizes nvcc for manywheel

(Tested in pytorch/pytorch#57522, sample logs https://app.circleci.com/pipelines/github/pytorch/pytorch/326891/workflows/b4a3ecdb-b340-4ccf-ac9b-a7c0b9174224/jobs/13691140)

if "%CUDA_VERSION%" == "110" call internal\driver_update.bat
if "%CUDA_VERSION%" == "111" call internal\driver_update.bat
if "%CUDA_VERSION%" == "112" call internal\driver_update.bat
if DEFINED CUDA_VERSION call internal\driver_update.bat
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Couldn't CUDA_VERSION also be set to cpu?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good call, I'll change this to check for inequality with cpu

@janeyx99 janeyx99 force-pushed the fix-windows-11.3 branch from 06ce889 to 161ff13 Compare May 7, 2021 16:13
@janeyx99 janeyx99 force-pushed the fix-windows-11.3 branch from 2af9731 to 5c8ed42 Compare May 18, 2021 16:17
@janeyx99 janeyx99 force-pushed the fix-windows-11.3 branch from d0e24a4 to a2d2ce3 Compare May 24, 2021 15:56
@janeyx99 janeyx99 merged commit 08ef2ad into pytorch:master May 24, 2021
@janeyx99 janeyx99 changed the title Add thrust_11.3 explicitly and remove bad quotes Add thrust_11.3 explicitly, remove bad quotes, parallelize nvcc May 24, 2021
@janeyx99 janeyx99 mentioned this pull request May 25, 2021
19 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants