Skip to content

Commit d37ea64

Browse files
committed
Exclude model generation for AGX
1 parent 0347755 commit d37ea64

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

qa/common/gen_qa_model_repository

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -430,7 +430,7 @@ python3 $VOLUME_SRCDIR/gen_qa_ragged_models.py --tensorrt --models_dir=$VOLUME_R
430430
chmod -R 777 $VOLUME_RAGGEDDESTDIR
431431
python3 $VOLUME_SRCDIR/gen_qa_trt_format_models.py --models_dir=$VOLUME_FORMATDESTDIR
432432
chmod -R 777 $VOLUME_FORMATDESTDIR
433-
python3 $VOLUME_SRCDIR/gen_qa_trt_data_dependent_shape.py --models_dir=$VOLUME_DATADEPENDENTDIR
433+
nvidia-smi --query-gpu=compute_cap | grep -qz 11.0 && echo -e '\033[33m[WARNING]\033[0m Skipping model generation for data dependent shape' || python3 $VOLUME_SRCDIR/gen_qa_trt_data_dependent_shape.py --models_dir=$VOLUME_DATADEPENDENTDIR
434434
chmod -R 777 $VOLUME_DATADEPENDENTDIR
435435
# Make shared library for custom Hardmax plugin.
436436
if [ -d "/usr/src/tensorrt" ]; then

0 commit comments

Comments
 (0)