Merge pull request #233057 from mweinelt/onnx-1.14.0

python310Packages.onnx: 0.13.1 -> 0.14.0
This commit is contained in:
Martin Weinelt 2023-05-21 15:24:01 +02:00 committed by GitHub
commit 8c881d076a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,55 +1,50 @@
{ lib
, stdenv
, buildPythonPackage
, python3
, bash
, cmake
, fetchFromGitHub
, gtest
, isPy27
, nbval
, numpy
, parameterized
, protobuf
, pybind11
, pytestCheckHook
, six
, pythonOlder
, tabulate
, typing-extensions
, pythonRelaxDepsHook
}:
let
gtestStatic = gtest.override { static = true; };
in buildPythonPackage rec {
pname = "onnx";
version = "1.13.1";
version = "1.14.0";
format = "setuptools";
disabled = isPy27;
disabled = pythonOlder "3.8";
src = fetchFromGitHub {
owner = pname;
repo = pname;
rev = "refs/tags/v${version}";
hash = "sha256-10MH23XpAv/uDW/2tRFGS2lKU8hnaNBwbIBIgVc7Jpk=";
hash = "sha256-f+s25Y/jGosaSdoZY6PE3j6pENkfDcD+IQndrbtuzWg=";
};
nativeBuildInputs = [
cmake
pythonRelaxDepsHook
pybind11
];
pythonRelaxDeps = [ "protobuf" ];
propagatedBuildInputs = [
protobuf
numpy
six
typing-extensions
];
nativeCheckInputs = [
nbval
parameterized
pytestCheckHook
tabulate
];
@ -72,7 +67,6 @@ in buildPythonPackage rec {
# 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"
'' + lib.optionalString doCheck ''
export CMAKE_ARGS+=" -Dgoogletest_STATIC_LIBRARIES=${gtestStatic}/lib/libgtest.a -Dgoogletest_INCLUDE_DIRS=${lib.getDev gtestStatic}/include"
export ONNX_BUILD_TESTS=1
'';
@ -89,14 +83,18 @@ in buildPythonPackage rec {
# The setup.py does all the configuration
dontUseCmakeConfigure = true;
doCheck = true;
preCheck = ''
export HOME=$(mktemp -d)
# detecting source dir as a python package confuses pytest
mv onnx/__init__.py onnx/__init__.py.hidden
'';
pytestFlagsArray = [ "onnx/test" "onnx/examples" ];
pytestFlagsArray = [
"onnx/test"
"onnx/examples"
];
disabledTests = [
# attempts to fetch data from web
"test_bvlc_alexnet_cpu"
@ -108,11 +106,25 @@ in buildPythonPackage rec {
"test_squeezenet_cpu"
"test_vgg19_cpu"
"test_zfnet512_cpu"
] ++ lib.optionals stdenv.isAarch64 [
# AssertionError: Output 0 of test 0 in folder
"test__pytorch_converted_Conv2d_depthwise_padded"
"test__pytorch_converted_Conv2d_dilated"
"test_dft"
"test_dft_axis"
# AssertionError: Mismatch in test 'test_Conv2d_depthwise_padded'
"test_xor_bcast4v4d"
# AssertionError: assert 1 == 0
"test_ops_tested"
];
disabledTestPaths = [
# Unexpected output fields from running code: {'stderr'}
"onnx/examples/np_array_tensorproto.ipynb"
];
__darwinAllowLocalNetworking = true;
postCheck = ''
# run "cpp" tests
.setuptools-cmake-build/onnx_gtests