diff --git a/pkgs/development/python-modules/apache-tvm-ffi/default.nix b/pkgs/development/python-modules/apache-tvm-ffi/default.nix new file mode 100644 index 000000000000..b4cd7aa67562 --- /dev/null +++ b/pkgs/development/python-modules/apache-tvm-ffi/default.nix @@ -0,0 +1,69 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + + # build-system + cmake, + cython, + ninja, + scikit-build-core, + setuptools-scm, + + # dependencies + typing-extensions, + + # tests + numpy, + pytestCheckHook, + writableTmpDirAsHomeHook, +}: + +buildPythonPackage (finalAttrs: { + pname = "apache-tvm-ffi"; + version = "0.1.9"; + pyproject = true; + + src = fetchFromGitHub { + owner = "apache"; + repo = "tvm-ffi"; + tag = "v${finalAttrs.version}"; + fetchSubmodules = true; + hash = "sha256-XnlM//WW2TbjbmzYBq6itJQ7R3J646UMVQUVhV5Afwc="; + }; + + build-system = [ + cmake + cython + ninja + scikit-build-core + setuptools-scm + ]; + dontUseCmakeConfigure = true; + + dependencies = [ + typing-extensions + ]; + + optional-dependencies = { + cpp = [ + ninja + ]; + }; + + pythonImportsCheck = [ "tvm_ffi" ]; + + nativeCheckInputs = [ + numpy + pytestCheckHook + writableTmpDirAsHomeHook + ]; + + meta = { + description = "Open ABI and FFI for Machine Learning Systems"; + changelog = "https://github.com/apache/tvm-ffi/releases/tag/${finalAttrs.src.tag}"; + homepage = "https://github.com/apache/tvm-ffi"; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ GaetanLepage ]; + }; +}) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index b6bb7a027f12..1d89c5e30dd5 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -834,6 +834,8 @@ self: super: with self; { apache-beam = callPackage ../development/python-modules/apache-beam { }; + apache-tvm-ffi = callPackage ../development/python-modules/apache-tvm-ffi { }; + apcaccess = callPackage ../development/python-modules/apcaccess { }; apeye = callPackage ../development/python-modules/apeye { };