chrishkchris edited a comment on issue #802:
URL: https://github.com/apache/singa/issues/802#issuecomment-727214676


   > > @naili-xing Naili, since you told us that you would like to contribute 
in SINGA, could you assist us to implement this (currently we have cpu test, 
but we also need GPU test)
   > 
   > Ok, i will look into it
   
   FYI, the different between CPU version and GPU version in building SINGA is 
just the flag we used in cmake
   GPU version: 
https://github.com/apache/singa/blob/master/tool/docker/devel/ubuntu/cuda10/Dockerfile#L104
   `cmake -DENABLE_TEST=ON -DUSE_CUDA=ON -DUSE_PYTHON3=ON -DUSE_DNNL=ON ..`
   
   CPU version: 
https://github.com/apache/singa/blob/master/tool/docker/devel/ubuntu/cpu/Dockerfile#L67
   `cmake -DENABLE_TEST=ON -DUSE_PYTHON3=ON -DUSE_DNNL=ON ..`
   
   In GPU version, it include all the GPU test case when we run 
test/python/run.py
   In CPU version, it automatically skiped all the GPU test case


----------------------------------------------------------------
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]


Reply via email to