Dev refactor generator (#5397)
* add random generator * reformat * refactor: allow auto generator * refactor: remove kAUTO, update python api and test * refactor: use member initializer lists, fix build issue when cpu only * handle exeception given invalid device * add dropout functor; add OpExprInterpContext; refactor random_mask_like based on random_generator * refactor random generator * Cast explicitily in unpack call to avoid confilt with Optional. * Refactor generator * Fix bad merge * Refine * Private default generator for each cuda device. * Fix * Refactor generator and refine unittest * add lock; refine testcase Co-authored-by:VertexC <bob2420083992@gmail.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/api/python/common.cpp 40 additions, 0 deletionsoneflow/api/python/common.cpp
- oneflow/api/python/common.h 31 additions, 0 deletionsoneflow/api/python/common.h
- oneflow/api/python/framework/device.cpp 5 additions, 13 deletionsoneflow/api/python/framework/device.cpp
- oneflow/api/python/framework/random_generator.cpp 13 additions, 5 deletionsoneflow/api/python/framework/random_generator.cpp
- oneflow/core/device/cuda_util.h 0 additions, 1 deletiononeflow/core/device/cuda_util.h
- oneflow/core/framework/random_generator.cpp 88 additions, 86 deletionsoneflow/core/framework/random_generator.cpp
- oneflow/core/framework/random_generator.h 35 additions, 135 deletionsoneflow/core/framework/random_generator.h
- oneflow/core/framework/random_generator_impl.cpp 114 additions, 0 deletionsoneflow/core/framework/random_generator_impl.cpp
- oneflow/core/framework/random_generator_impl.cu 42 additions, 0 deletionsoneflow/core/framework/random_generator_impl.cu
- oneflow/core/framework/random_generator_impl.h 167 additions, 0 deletionsoneflow/core/framework/random_generator_impl.h
- oneflow/core/functional/impl/nn_functor.cpp 1 addition, 1 deletiononeflow/core/functional/impl/nn_functor.cpp
- oneflow/python/framework/generator.py 9 additions, 5 deletionsoneflow/python/framework/generator.py
- oneflow/python/test/generator/test_generator.py 50 additions, 3 deletionsoneflow/python/test/generator/test_generator.py
- oneflow/user/kernels/conv_kernels.cpp 1 addition, 1 deletiononeflow/user/kernels/conv_kernels.cpp
- oneflow/user/kernels/random_mask_generator.cpp 1 addition, 4 deletionsoneflow/user/kernels/random_mask_generator.cpp
- oneflow/user/kernels/random_mask_generator.cu 3 additions, 4 deletionsoneflow/user/kernels/random_mask_generator.cu
- oneflow/user/kernels/random_mask_generator.h 8 additions, 4 deletionsoneflow/user/kernels/random_mask_generator.h
- oneflow/user/kernels/random_mask_like_kernel.h 3 additions, 1 deletiononeflow/user/kernels/random_mask_like_kernel.h
Please register or sign in to comment