diff options
Diffstat (limited to '')
| -rw-r--r-- | misc/py-pytorch/Makefile | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/misc/py-pytorch/Makefile b/misc/py-pytorch/Makefile index 4713de8cc9e9..227e1ea21933 100644 --- a/misc/py-pytorch/Makefile +++ b/misc/py-pytorch/Makefile @@ -1,7 +1,7 @@ PORTNAME= pytorch DISTVERSIONPREFIX= v DISTVERSION= 2.8.0 -PORTREVISION= 1 +PORTREVISION= 2 CATEGORIES= misc # machine-learning MASTER_SITES= https://github.com/pytorch/pytorch/releases/download/v${DISTVERSION}/ PKGNAMEPREFIX= ${PYTHON_PKGNAMEPREFIX} @@ -72,6 +72,14 @@ POST_PLIST= fix-plist TEST_ENV= ${MAKE_ENV} PYTHONPATH=${STAGEDIR}${PYTHONPREFIX_SITELIBDIR} TEST_WRKSRC= ${WRKSRC}/test +OPTIONS_DEFINE= VULKAN +OPTIONS_DEFAULT= VULKAN # VULKAN is experimental in PyTorch and only simple computations work + +VULKAN_MAKE_ENV= USE_VULKAN=1 USE_VULKAN_SHADERC_RUNTIME=1 USE_VULKAN_WRAPPER=0 # see https://docs.pytorch.org/tutorials/unstable/vulkan_workflow.html +VULKAN_BUILD_DEPENDS= glslc:graphics/shaderc \ + vulkan-headers>0:graphics/vulkan-headers +VULKAN_LIB_DEPENDS= libvulkan.so:graphics/vulkan-loader + .include <bsd.port.options.mk> .if ${OPSYS} == FreeBSD |