diff mbox series

[bug#63991,v2,2/3] gnu: Add dmlc-dlpack

Message ID 16d1ba48cab9b74bb9a39560606409008b68d02e.1686980668.git.atai@atai.org
State New
Headers show
Series [bug#63991,v2,1/3] gnu: Add rang | expand

Commit Message

Andy Tai June 17, 2023, 5:45 a.m. UTC 
* gnu/packages/machine-learning.scm (dmlc-dlpack): New variable
---
 gnu/packages/machine-learning.scm | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
diff mbox series

Patch

diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm
index 927b531cf2..ac115fc2ad 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -4584,3 +4584,27 @@  (define-public oneapi-dnnl
      "OneAPI Deep Neural Network Library (oneDNN) is a cross-platform
 performance library of basic building blocks for deep learning applications.")
     (license license:asl2.0)))
+
+
+(define-public dmlc-dlpack
+  (package
+    (name "dmlc-dlpack")
+    (version "0.8")
+    (source
+     (origin
+       (method git-fetch)
+       (uri (git-reference
+             (url "https://github.com/dmlc/dlpack")
+             (commit (string-append "v" version))))
+       (file-name (git-file-name name version))
+       (sha256
+        (base32 "1jmfvdrx8dn9b0ksm04ixha0jnrk35ip796qg98kfz6g7niw5ir1"))))
+    (build-system cmake-build-system)
+    (arguments (list #:tests? #f))  ; no tests
+    (home-page "https://dmlc.github.io/dlpack/latest")
+    (synopsis "Open In Memory Tensor Structure")
+    (description
+     "DLPack is an open in-memory tensor structure for sharing tensors among frameworks")
+    (license license:asl2.0)))
+
+