Fix 'custom_ops' test configuration (#543)

* Update '--preload-path' value

* Fix docker mnt volumes when LD_PRELOAD has colon

Co-authored-by: Iman Tabrizian <[email protected]>

Author: mc-nv

qa/L0_custom_ops/test.sh

@@ -34,7 +34,7 @@ TRITON_LOG_BASE="triton.log"
 WAIT_TIMEOUT=1200
 
 # Generate test configs
-python3 test_config_generator.py --profile-models $MODEL_NAMES --preload-path $MODEL_REPOSITORY/libtorch_modulo/custom_modulo.so --library-path /opt/tritonserver/backends/pytorch:'$LD_LIBRARY_PATH'
+python3 test_config_generator.py --profile-models $MODEL_NAMES --preload-path "/usr/lib/x86_64-linux-gnu/libpython3.8.so.1:$MODEL_REPOSITORY/libtorch_modulo/custom_modulo.so" --library-path /opt/tritonserver/backends/pytorch:'$LD_LIBRARY_PATH'
 
 LIST_OF_CONFIG_FILES=(`ls | grep .yaml`)
 

qa/L0_custom_ops/test_config_generator.py

@@ -94,8 +94,11 @@ def generate_docker_mode_custom_op_config(self):
         if 'TRITON_LAUNCH_DOCKER_IMAGE' in os.environ:
             self.config['triton_docker_image'] = os.environ[
                 'TRITON_LAUNCH_DOCKER_IMAGE']
+        preload_paths = self.args.preload_path.split(':')
+        # Only mount the second preload location. The first
+        # location is available in the server container.
         self.config['triton_docker_mounts'] = [
-            f'{self.args.preload_path}:{self.args.preload_path}:ro',
+            f'{preload_paths[1]}:{preload_paths[1]}:ro'
         ]
         self.config['triton_server_environment'] = {
             'LD_PRELOAD': self.args.preload_path,