Skip to content

Commit 04e1437

Browse files
committed
enable torch.backends tests also for win+CUDA
1 parent 381fcb8 commit 04e1437

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

recipe/meta.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -399,10 +399,10 @@ outputs:
399399
- python -c "import torch; import numpy"
400400
- python -c "import numpy; import torch"
401401
# distributed support is enabled by default on linux; for mac, we enable it manually in build.sh
402-
- python -c "import torch; assert torch.distributed.is_available()" # [linux or osx]
403-
- python -c "import torch; assert torch.backends.cuda.is_built()" # [linux64 and (cuda_compiler_version != "None")]
404-
- python -c "import torch; assert torch.backends.cudnn.is_available()" # [linux64 and (cuda_compiler_version != "None")]
405-
- python -c "import torch; assert torch.backends.cudnn.enabled" # [linux64 and (cuda_compiler_version != "None")]
402+
- python -c "import torch; assert torch.distributed.is_available()" # [linux or osx]
403+
- python -c "import torch; assert torch.backends.cuda.is_built()" # [cuda_compiler_version != "None"]
404+
- python -c "import torch; assert torch.backends.cudnn.is_available()" # [cuda_compiler_version != "None"]
405+
- python -c "import torch; assert torch.backends.cudnn.enabled" # [cuda_compiler_version != "None"]
406406
- python -c "import torch; assert torch.version.cuda is not None" # [cuda_compiler_version != "None"]
407407
# At conda-forge, we target versions of OSX that are too old for MPS support
408408
# But if users install a newer version of OSX, they will have MPS support

0 commit comments

Comments
 (0)