Skip to content
Snippets Groups Projects
Unverified Commit b05dd645 authored by Houjiang Chen's avatar Houjiang Chen Committed by GitHub
Browse files

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: default avatarVertexC <bob2420083992@gmail.com>
Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
parent bc475cbe
No related branches found
No related tags found
No related merge requests found
Showing
with 611 additions and 263 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment