-
-
Notifications
You must be signed in to change notification settings - Fork 14.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #101728 from adelbertc/init-tvm
tvm: init at 0.7.0
- Loading branch information
Showing
2 changed files
with
32 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -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 ]; | ||
}; | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters