Merge pull request #101728 from adelbertc/init-tvm

tvm: init at 0.7.0

1 files changed, 30 insertions, 0 deletions

diff --git a/pkgs/development/compilers/tvm/default.nix b/pkgs/development/compilers/tvm/default.nix
new file mode 100644
index 0000000000000..01c44718f11a2
--- /dev/null
+++ b/pkgs/development/compilers/tvm/default.nix
@@ -0,0 +1,30 @@
+{ stdenv, fetchFromGitHub, cmake }:
+
+stdenv.mkDerivation rec {
+  pname = "tvm";
+  version = "0.7.0";
+
+  src = fetchFromGitHub {
+    owner = "apache";
+    repo = "incubator-tvm";
+    rev = "v${version}";
+    fetchSubmodules = true;
+    sha256 = "0qflpd3lw0jslyk5lqpv2v42lkqs8mkvnn6i3fdms32iskdfk6p5";
+  };
+
+  nativeBuildInputs = [ cmake ];
+  # TVM CMake build uses some sources in the project's ./src/target/opt/
+  # directory which errneously gets mangled by the eager `fixCmakeFiles`
+  # function in Nix's CMake setup-hook.sh to ./src/target/var/empty/,
+  # which then breaks the build. Toggling this flag instructs Nix to
+  # not mangle the legitimate use of the opt/ folder.
+  dontFixCmake = true;
+
+  meta = with stdenv.lib; {
+    homepage = "https://tvm.apache.org/";
+    description = "An End to End Deep Learning Compiler Stack for CPUs, GPUs and accelerators";
+    license = licenses.asl20;
+    platforms = platforms.all;
+    maintainers = with maintainers; [ adelbertc ];
+  };
+}