taco: nixfmt
This commit is contained in:
@@ -1,15 +1,23 @@
|
||||
{ stdenv
|
||||
, lib
|
||||
, fetchgit
|
||||
, cmake
|
||||
, llvmPackages
|
||||
, enablePython ? false
|
||||
, python ? null
|
||||
{
|
||||
stdenv,
|
||||
lib,
|
||||
fetchgit,
|
||||
cmake,
|
||||
llvmPackages,
|
||||
enablePython ? false,
|
||||
python ? null,
|
||||
}:
|
||||
|
||||
let pyEnv = python.withPackages (p: with p; [ numpy scipy ]);
|
||||
let
|
||||
pyEnv = python.withPackages (
|
||||
p: with p; [
|
||||
numpy
|
||||
scipy
|
||||
]
|
||||
);
|
||||
|
||||
in stdenv.mkDerivation rec {
|
||||
in
|
||||
stdenv.mkDerivation rec {
|
||||
pname = "taco";
|
||||
version = "unstable-2022-08-02";
|
||||
|
||||
@@ -31,7 +39,7 @@ in stdenv.mkDerivation rec {
|
||||
|
||||
cmakeFlags = [
|
||||
"-DOPENMP=ON"
|
||||
] ++ lib.optional enablePython "-DPYTHON=ON" ;
|
||||
] ++ lib.optional enablePython "-DPYTHON=ON";
|
||||
|
||||
postInstall = lib.strings.optionalString enablePython ''
|
||||
mkdir -p $out/${python.sitePackages}
|
||||
@@ -45,11 +53,11 @@ in stdenv.mkDerivation rec {
|
||||
# However, the python module works flawlessly.
|
||||
dontFixup = enablePython;
|
||||
|
||||
meta = with lib; {
|
||||
meta = {
|
||||
description = "Computes sparse tensor expressions on CPUs and GPUs";
|
||||
mainProgram = "taco";
|
||||
license = licenses.mit;
|
||||
license = lib.licenses.mit;
|
||||
homepage = "https://github.com/tensor-compiler/taco";
|
||||
maintainers = [ maintainers.sheepforce ];
|
||||
maintainers = [ lib.maintainers.sheepforce ];
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user