gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cudaPackages.tests.onnx-tensorrt: init

Browse Source 
Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
This commit is contained in:
Connor Baker
2025-10-22 13:14:49 -07:00
parent d027f0a543
commit 71228337e5
2 changed files with 83 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/cuda-modules/packages/tests/onnx-tensorrt/long.nix
+40
View File
@@ -0,0 +1,40 @@
{
cuda_cudart,
cudaNamePrefix,
onnx-tensorrt,
python3,
writeShellApplication,
}:
writeShellApplication {
derivationArgs = {
__structuredAttrs = true;
strictDeps = true;
};
name = "${cudaNamePrefix}-tests-onnx-tensorrt-long";
runtimeInputs = [
cuda_cudart
(python3.withPackages (ps: [
ps.onnx-tensorrt
ps.pytest
ps.six
]))
];
text = ''
args=(
python3
"${onnx-tensorrt.test_script}/onnx_backend_test.py"
)
if (( $# != 0 ))
then
args+=( "$@" )
else
args+=( --verbose )
echo "Running with default arguments: ''${args[*]}" >&2
fi
mkdir -p "$HOME/.onnx"
chmod -R +w "$HOME/.onnx"
"''${args[@]}"
'';
}
pkgs/development/cuda-modules/packages/tests/onnx-tensorrt/short.nix
+43
View File
@@ -0,0 +1,43 @@
{
cuda_cudart,
cudaNamePrefix,
onnx-tensorrt,
python3,
writeShellApplication,
}:
writeShellApplication {
derivationArgs = {
__structuredAttrs = true;
strictDeps = true;
};
name = "${cudaNamePrefix}-tests-onnx-tensorrt-short";
runtimeInputs = [
cuda_cudart
(python3.withPackages (ps: [
ps.onnx-tensorrt
ps.pytest
ps.six
]))
];
text = ''
args=(
python3
"${onnx-tensorrt.test_script}/onnx_backend_test.py"
)
if (( $# != 0 ))
then
args+=( "$@" )
else
args+=(
--verbose
OnnxBackendRealModelTest
)
echo "Running with default arguments: ''${args[*]}" >&2
fi
mkdir -p "$HOME/.onnx"
chmod -R +w "$HOME/.onnx"
"''${args[@]}"
'';
}