samskalicky edited a comment on issue #17270: Dynamic custom operator GPU 
support
URL: https://github.com/apache/incubator-mxnet/pull/17270#issuecomment-578608204
 
 
   @rondogency if you're going to add a new library for the GPU tests you need 
to modify: CMakeLists.txt, Jenkins_steps.groovy. In the CI cmake is used, and 
you need to store the library between build/test stages of the CI by modifying 
the groovy file. 
   
   See the subgraph property PR modifications for these files that add 
`libsubgraph_lib.so`:
   https://github.com/apache/incubator-mxnet/pull/17034/files
   
   At least the addition of the `__attribute__ ((visibility ("hidden")))` fixed 
the multiple library symbol problem:
   ```
   test_extensions.test_custom_op ... [02:06:10] src/c_api/c_api.cc:286: Found 
2 operators in library
   [02:06:10] src/c_api/c_api.cc:350:   Op[0] my_gemm
   [02:06:10] src/c_api/c_api.cc:350:   Op[1] state_gemm
   [02:06:10] src/c_api/c_api.cc:785: Found 0 partitioners in library
   Info: stateful operator created
   Info: keyword + number of forward: 1
   Info: keyword + number of forward: 2
   ok (0.5803s)
   test_extensions.test_custom_op_gpu ... SKIP: ignoring custom_op_gpu test on 
cpu run
   MXNet version 10600 supported
   test_extensions.test_subgraph ... [02:06:11] src/c_api/c_api.cc:286: Found 1 
operators in library
   [02:06:11] src/c_api/c_api.cc:350:   Op[0] _custom_subgraph_op
   [02:06:11] src/c_api/c_api.cc:785: Found 1 partitioners in library
   [02:06:11] src/c_api/c_api.cc:801:   Partitioner[0] myProp
   [02:06:11] src/c_api/c_api.cc:821:           Strategy[0] strategy1 
subgraphOp: '_custom_subgraph_op'
   ```
   

----------------------------------------------------------------
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:
[email protected]


With regards,
Apache Git Services

Reply via email to