Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.3 kB · 122 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123{ lib, buildPythonPackage, fetchFromGitHub,
# build-system cmake, pybind11, setuptools,
# buildInputs abseil-cpp, protobuf, gtest,
# dependencies ml-dtypes, numpy, typing-extensions,
# tests google-re2, nbval, parameterized, pillow, pytestCheckHook, tabulate, writableTmpDirAsHomeHook,}:
let gtestStatic = gtest.override { static = true; };inbuildPythonPackage rec { pname = "onnx"; version = "1.19.0"; pyproject = true;
src = fetchFromGitHub { owner = "onnx"; repo = "onnx"; tag = "v${version}"; hash = "sha256-dDc7ugzQHcArf9TRcF9Ofv16jc3gqhMWCZrYKJ7Udfw="; };
build-system = [ cmake protobuf setuptools ];
buildInputs = [ abseil-cpp gtestStatic pybind11 ];
dependencies = [ ml-dtypes numpy protobuf typing-extensions ];
nativeCheckInputs = [ google-re2 ml-dtypes nbval parameterized pillow pytestCheckHook tabulate writableTmpDirAsHomeHook ];
preConfigure = '' # Set CMAKE_INSTALL_LIBDIR to lib explicitly, because otherwise it gets set # to lib64 and cmake incorrectly looks for the protobuf library in lib64 export CMAKE_ARGS="-DCMAKE_INSTALL_LIBDIR=lib -DONNX_USE_PROTOBUF_SHARED_LIBS=ON" export CMAKE_ARGS+=" -Dgoogletest_STATIC_LIBRARIES=${gtestStatic}/lib/libgtest.a" export ONNX_BUILD_TESTS=1 '';
preBuild = '' export MAX_JOBS=$NIX_BUILD_CORES '';
# The executables are just utility scripts that aren't too important postInstall = '' rm -r $out/bin '';
# The setup.py does all the configuration dontUseCmakeConfigure = true;
# detecting source dir as a python package confuses pytest preCheck = '' rm onnx/__init__.py '';
enabledTestPaths = [ "onnx/test" "examples" ];
__darwinAllowLocalNetworking = true;
postCheck = '' # run "cpp" tests .setuptools-cmake-build/onnx_gtests '';
pythonImportsCheck = [ "onnx" ];
meta = { description = "Open Neural Network Exchange"; homepage = "https://onnx.ai"; changelog = "https://github.com/onnx/onnx/releases/tag/v${version}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ acairncross ]; };}