From e4683ce6184e3f5a088adc364cd347f81d65c578 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Tue, 7 Jan 2025 13:16:58 -0800 Subject: [PATCH 1/4] Update variable name --- qa/common/gen_qa_model_repository | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qa/common/gen_qa_model_repository b/qa/common/gen_qa_model_repository index 73dfee3189..8538e59a74 100755 --- a/qa/common/gen_qa_model_repository +++ b/qa/common/gen_qa_model_repository @@ -286,7 +286,7 @@ python3 $VOLUME_SRCDIR/gen_qa_dyna_sequence_implicit_models.py --onnx --onnx_ops chmod -R 777 $VOLUME_DYNASEQIMPLICITDESTDIR python3 $VOLUME_SRCDIR/gen_qa_ragged_models.py --onnx --onnx_opset=$ONNX_OPSET --models_dir=$VOLUME_RAGGEDDESTDIR chmod -R 777 $VOLUME_RAGGEDDESTDIR -python3 $VOLUME_SRCDIR/gen_qa_ort_scalar_models.py --onnx_opset=$ONNX_OPSET --models_dir=$SCALARMODELSDESTDIR +python3 $VOLUME_SRCDIR/gen_qa_ort_scalar_models.py --onnx_opset=$ONNX_OPSET --models_dir=$VOLUME_SCALARMODELSDESTDIR chmod -R 777 $VOLUME_RAGGEDDESTDIR EOF From 5de981625344d63110f4e4555152af58a5017f9b Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Tue, 7 Jan 2025 13:31:42 -0800 Subject: [PATCH 2/4] set permissions to the folder --- qa/common/gen_qa_model_repository | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qa/common/gen_qa_model_repository b/qa/common/gen_qa_model_repository index 8538e59a74..93e4dc2dfd 100755 --- a/qa/common/gen_qa_model_repository +++ b/qa/common/gen_qa_model_repository @@ -287,7 +287,7 @@ chmod -R 777 $VOLUME_DYNASEQIMPLICITDESTDIR python3 $VOLUME_SRCDIR/gen_qa_ragged_models.py --onnx --onnx_opset=$ONNX_OPSET --models_dir=$VOLUME_RAGGEDDESTDIR chmod -R 777 $VOLUME_RAGGEDDESTDIR python3 $VOLUME_SRCDIR/gen_qa_ort_scalar_models.py --onnx_opset=$ONNX_OPSET --models_dir=$VOLUME_SCALARMODELSDESTDIR -chmod -R 777 $VOLUME_RAGGEDDESTDIR +chmod -R 777 $VOLUME_SCALARMODELSDESTDIR EOF chmod a+x $ONNXSCRIPT From a254b9ca8ce9a8317cd12a9f499c9abd9327cc07 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Tue, 7 Jan 2025 13:46:36 -0800 Subject: [PATCH 3/4] Add exception if misesd argument value --- qa/common/gen_qa_ort_scalar_models.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/qa/common/gen_qa_ort_scalar_models.py b/qa/common/gen_qa_ort_scalar_models.py index f2ddb35912..f378e3461f 100755 --- a/qa/common/gen_qa_ort_scalar_models.py +++ b/qa/common/gen_qa_ort_scalar_models.py @@ -124,6 +124,10 @@ def create_onnx_modelconfig(models_dir, dtype, shape): ) FLAGS = parser.parse_args() + + if not FLAGS.models_dir: + raise Exception("--models_dir is required") + create_onnx_modelfile(FLAGS.models_dir, shape=[1], dtype=np.float32) create_onnx_modelconfig(FLAGS.models_dir, shape=[1], dtype=np.float32) create_onnx_modelfile(FLAGS.models_dir, shape=[1, 1], dtype=np.float32) From 96fc61d80f1a0731229393ae6510096f463e4959 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Tue, 7 Jan 2025 13:49:56 -0800 Subject: [PATCH 4/4] Fix copyright header --- qa/common/gen_qa_ort_scalar_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qa/common/gen_qa_ort_scalar_models.py b/qa/common/gen_qa_ort_scalar_models.py index f378e3461f..c00a97d5ed 100755 --- a/qa/common/gen_qa_ort_scalar_models.py +++ b/qa/common/gen_qa_ort_scalar_models.py @@ -1,6 +1,6 @@ #!/usr/bin/env python3 -# Copyright 2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright 2023-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions