Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.4 kB · 64 lines
Nix
at litex
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465{ lib, buildPythonPackage, autoPatchelfHook, pythonRelaxDepsHook, onnxruntime, coloredlogs, numpy, packaging, oneDNN
}:
# onnxruntime requires an older protobuf.# Doing an override in protobuf in the python-packages set# can give you a functioning Python package but note not# all Python packages will be compatible then.## Because protobuf is not always needed we remove it# as a runtime dependency from our wheel.## We do include here the non-Python protobuf so the shared libs# link correctly. If you do also want to include the Python# protobuf, you can add it to your Python env, but be aware# the version likely mismatches with what is used here.
buildPythonPackage { inherit (onnxruntime) pname version; format = "wheel"; src = onnxruntime.dist;
unpackPhase = '' cp -r $src dist chmod +w dist '';
nativeBuildInputs = [ autoPatchelfHook pythonRelaxDepsHook ];
# This project requires fairly large dependencies such as sympy which we really don't always need. pythonRemoveDeps = [ "flatbuffers" "protobuf" "sympy" ];
# Libraries are not linked correctly. buildInputs = [ oneDNN onnxruntime.protobuf ];
propagatedBuildInputs = [ coloredlogs # flatbuffers numpy packaging # protobuf # sympy ];
meta = onnxruntime.meta // { maintainers = with lib.maintainers; [ fridh ]; };}