[GitHub] [incubator-mxnet] ptrendx commented on a change in pull request #18622: [WIP] Use RTC for elementwise and broadcast ops

[GitBox <gi...@apache.org>]

ptrendx commented on a change in pull request #18622:
URL: https://github.com/apache/incubator-mxnet/pull/18622#discussion_r457584073



##########
File path: ci/docker/Dockerfile.build.jetson
##########
@@ -90,6 +90,9 @@ RUN wget https://repo.download.nvidia.com/jetson/x86_64/pool/r32.4/c/cuda/cuda-c
 # while previously it used to store it at /usr/local/cuda/targets/aarch64-linux/lib/stubs
 RUN ln -s /usr/lib/aarch64-linux-gnu/libcublas.so /usr/local/cuda/targets/aarch64-linux/lib/stubs/libcublas.so
 
+# Need the stub for libcuda.so in a place where host linker can find it to link the unit tests
+RUN ln -s /usr/local/cuda/targets/aarch64-linux/lib/stubs/libcuda.so /usr/lib/aarch64-linux-gnu/libcuda.so

Review comment:
       It is detected by cmake, but only for CUDA compilation - the problem happens during the compilation of mxnet_unit_tests, which is not CUDA-aware I believe.




----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org