Giters

NervanaSystems / ngraph-mxnet

MXNet - nGraph integration

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Enable nGraph GPU transformer for Ngraph_Mxnet

lnguyen-nvn opened this issue · comments

commented

Enable nGraph GPU transformer for Ngraph_Mxnet is stretch goal Q4.

  1. I tested the style checks, all test suites in the directory test/python/gpu/ with nvidia/cuda:9.1, cudnn 7, python 3, and found out the following issues:

https://jira01.devtools.intel.com/browse/NGRAPH-3118 : Failed to execute test_operator_gpu.py (Core dump)
https://jira01.devtools.intel.com/browse/NGRAPH-3198 : Disable test_tvm_bridge.py

  1. Compling the Ngraph_Mxnet with GPU is taken agv 22 mins ( ~ double times if comparing with CPU_MKLDNN). I am not sure if we can optimize /speed up the timing for compilation.

  2. Please provide OS need to cover for nGPU ( CUDA version, CuDNN version).