tjh.dev
1{ lib, stdenv, fetchFromGitHub, cmake }:
2
3stdenv.mkDerivation rec {
4 pname = "tvm";
5 version = "0.12.0";
6
7 src = fetchFromGitHub {
8 owner = "apache";
9 repo = "incubator-tvm";
10 rev = "v${version}";
11 fetchSubmodules = true;
12 sha256 = "sha256-NHfYx45Zad+jsILR24c2U+Xmb2rKaTyl8xl5uxAFtak=";
13 };
14
15 nativeBuildInputs = [ cmake ];
16 # TVM CMake build uses some sources in the project's ./src/target/opt/
17 # directory which errneously gets mangled by the eager `fixCmakeFiles`
18 # function in Nix's CMake setup-hook.sh to ./src/target/var/empty/,
19 # which then breaks the build. Toggling this flag instructs Nix to
20 # not mangle the legitimate use of the opt/ folder.
21 dontFixCmake = true;
22
23 meta = with lib; {
24 homepage = "https://tvm.apache.org/";
25 description = "An End to End Deep Learning Compiler Stack for CPUs, GPUs and accelerators";
26 license = licenses.asl20;
27 platforms = platforms.all;
28 maintainers = with maintainers; [ adelbertc ];
29 };
30}