diff mbox series

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

Message ID 33d6ff99a3580f99d25333da94140dadd291e8cb.1686380735.git.atai@atai.org
State New
Headers show
Series Add Apache Twm and dependencies | expand

Commit Message

Andy Tai June 10, 2023, 7:12 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 c9bc5e8b68..4ed32a8792 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -4388,3 +4388,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)))
+
+