Inconsistent return types of random_generator functions in cpu and gpu #13187
Replies: 3 comments
-
|
Hello @ChaiBapchya @mxnet-label-bot add [Question] |
Beta Was this translation helpful? Give feedback.
-
|
@mxnet-label-bot add [Question] |
Beta Was this translation helpful? Give feedback.
-
|
There is a template specialization for double in gpu. Generic implementation without specialization is not so simple because the underlying curand API is not generic. It requires an implementation of a new generic random number generator. Actually I have tried to address some issues in random number generation by implementing a new generator common for cpu and gpu (dmlc/mshadow#338), but no one might be interested in the proposal 😁 |
Beta Was this translation helpful? Give feedback.
Uh oh!
There was an error while loading. Please reload this page.
-
The file -
/incubator-mxnet/include/mxnet/random_generator.hincludes CPU and GPU implementation of random number generator functions.Close look at
cpuimplementationWhile in case of
gpuWhy is it hardcoded in gpu? Shouldn't it be generic?
Beta Was this translation helpful? Give feedback.
All reactions