feat: Added env var for TOP_DIR

Signed-off-by: Anurag Dixit <[email protected]>
pytorch · Nov 30, 2021 · c26180e · c26180e
1 parent 161ef3d
commit c26180e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/dist-accuracy-test.sh b/docker/dist-accuracy-test.sh
@@ -15,7 +15,7 @@ chmod a+x /usr/bin/bazel
 export NVIDIA_TF32_OVERRIDE=0
 
 cd /opt/pytorch/torch_tensorrt
-cp cp /opt/pytorch/torch_tensorrt/docker/WORKSPACE.docker  /opt/pytorch/torch_tensorrt/WORKSPACE
+cp /opt/pytorch/torch_tensorrt/docker/WORKSPACE.docker  /opt/pytorch/torch_tensorrt/WORKSPACE
 
 pip install --user --upgrade nox
-nox 
+TOP_DIR=/opt/pytorch/torch_tensorrt nox