Skip to content
Snippets Groups Projects
Commit 5aeab7db authored by Glenn Johnson's avatar Glenn Johnson Committed by Adam J. Stewart
Browse files

Use CudaPackage mixin for py-torch (#14540)

This PR adds CudaPackage in order to pick up the cuda/compiler conflicts
defined in CudaPackage.
parent 8ad0be96
No related branches found
No related tags found
No related merge requests found
...@@ -7,7 +7,7 @@ ...@@ -7,7 +7,7 @@
# TODO: try switching to CMakePackage for more control over build # TODO: try switching to CMakePackage for more control over build
class PyTorch(PythonPackage): class PyTorch(PythonPackage, CudaPackage):
"""Tensors and Dynamic neural networks in Python """Tensors and Dynamic neural networks in Python
with strong GPU acceleration.""" with strong GPU acceleration."""
...@@ -61,7 +61,6 @@ class PyTorch(PythonPackage): ...@@ -61,7 +61,6 @@ class PyTorch(PythonPackage):
version('0.4.0', tag='v0.4.0', submodules=True) version('0.4.0', tag='v0.4.0', submodules=True)
version('0.3.1', tag='v0.3.1', submodules=True) version('0.3.1', tag='v0.3.1', submodules=True)
variant('cuda', default=True, description='Enables CUDA build')
variant('cudnn', default=True, description='Enables the cuDNN build') variant('cudnn', default=True, description='Enables the cuDNN build')
variant('magma', default=False, description='Enables the MAGMA build') variant('magma', default=False, description='Enables the MAGMA build')
variant('fbgemm', default=False, description='Enables the FBGEMM build') variant('fbgemm', default=False, description='Enables the FBGEMM build')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment