diff options
| author | Adelbert Chang <adelbertc@gmail.com> | 2020-10-25 20:37:33 -0700 |
|---|---|---|
| committer | Adelbert Chang <adelbertc@gmail.com> | 2020-11-26 21:53:17 -0800 |
| commit | 2eafe0ca1c30e43e73fe775859da1c15bc709f0d (patch) | |
| tree | 3d13cb44961c58c5e0ba8141185b3bf98cf541a5 | |
| parent | 3216b8571373b368a0ce809e2976fad8bffd0b90 (diff) | |
| download | nixpkgs-2eafe0ca1c30e43e73fe775859da1c15bc709f0d.tar nixpkgs-2eafe0ca1c30e43e73fe775859da1c15bc709f0d.tar.gz nixpkgs-2eafe0ca1c30e43e73fe775859da1c15bc709f0d.tar.bz2 nixpkgs-2eafe0ca1c30e43e73fe775859da1c15bc709f0d.tar.lz nixpkgs-2eafe0ca1c30e43e73fe775859da1c15bc709f0d.tar.xz nixpkgs-2eafe0ca1c30e43e73fe775859da1c15bc709f0d.tar.zst nixpkgs-2eafe0ca1c30e43e73fe775859da1c15bc709f0d.zip | |
tvm: init at 0.7.0
| -rw-r--r-- | pkgs/development/compilers/tvm/default.nix | 30 | ||||
| -rw-r--r-- | pkgs/top-level/all-packages.nix | 2 |
2 files changed, 32 insertions, 0 deletions
diff --git a/pkgs/development/compilers/tvm/default.nix b/pkgs/development/compilers/tvm/default.nix new file mode 100644 index 00000000000..01c44718f11 --- /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 ]; + }; +} diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 9efc75240c0..c5e7358ce10 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -7601,6 +7601,8 @@ in turses = callPackage ../applications/networking/instant-messengers/turses { }; + tvm = callPackage ../development/compilers/tvm { }; + oysttyer = callPackage ../applications/networking/instant-messengers/oysttyer { }; twilight = callPackage ../tools/graphics/twilight { |