Giters

rocmarchive / cunn_hip

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Enable hipRNG for RELU

bensander opened this issue · comments

commented

You will need to uncomment the code in lib/THCUNN/RReLU.cu that uses RNG, change it to use hiprng, and other associated cleanup
o If successful, we should see these link warning disappear:

root@3c838b17492a:~/cunn_hip# luajit -l cunn -e 'cunn.test()'
not found: THNN_CudaRReLU_updateOutput/root/torch/install/share/lua/5.1/nn/THNN.lua:108: /root/torch/install/lib/lua/5.1/libTHCUNN.so: undefined symbol: THNN_CudaRReLU_updateOutput
not found: THNN_CudaRReLU_updateGradInput/root/torch/install/share/lua/5.1/nn/THNN.lua:108: /root/torch/install/lib/lua/5.1/libTHCUNN.so: undefined symbol: THNN_CudaRReLU_updateGradInput

commented

87b9f38 has the fix for it.

The checkin however needs to get validated. But the undefined reference is resolved however