zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-onnxruntime: use CudaPackage (#47684)

Browse Source
This commit is contained in:
Wouter Deconinck 2025-03-17 04:35:20 -05:00 committed by GitHub
parent 0f84623914
commit e15a530f32
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
var/spack/repos/builtin/packages/py-onnxruntime/package.py
View File

@ -6,7 +6,7 @@
from spack.package import *
class PyOnnxruntime(CMakePackage, PythonExtension, ROCmPackage):
class PyOnnxruntime(CMakePackage, PythonExtension, ROCmPackage, CudaPackage):
"""ONNX Runtime is a performance-focused complete scoring
engine for Open Neural Network Exchange (ONNX) models, with
an open extensible architecture to continually address the
@ -35,8 +35,6 @@ class PyOnnxruntime(CMakePackage, PythonExtension, ROCmPackage):
depends_on("c", type="build") # generated
depends_on("cxx", type="build") # generated
variant("cuda", default=False, description="Build with CUDA support")
# cmake/CMakeLists.txt
depends_on("cmake@3.26:", when="@1.17:", type="build")
depends_on("cmake@3.1:", type="build")