mxnet-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From Leonard Lausen <>
Subject Re: [apache/incubator-mxnet] [RFC] MXNet external operators (#18904)
Date Tue, 25 Aug 2020 01:58:50 GMT
@leezu commented on this pull request.

> @@ -33,6 +33,14 @@
 # Specify `cmake --build . --parallel N` to set the number of parallel compilation jobs.
 # Default is derived from CPUs available.
+# By default, cmake will try and discover which GPU architecture to use by looking at
+# the available GPUs on the machine that you're building on. If you want to build for
+# a specific GPU architecture or are building on a machine without a GPU, then
+# specify the MXNET_CUDA_ARCH option like:
+# $ cmake .. -DMXNET_CUDA_ARCH=7.0
+# In the example above we're building for sm_70 which is the Volta architecture.

To avoid duplication, should we just reference the option and explanation 20 lines below.If
you find the explanation below isn't clear, feel free to improve it. It may be better than
documenting it twice. WDYT? 

You are receiving this because you were mentioned.
Reply to this email directly or view it on GitHub:
  • Unnamed multipart/alternative (inline, 7-Bit, 0 bytes)
View raw message