Priority = P0good first issue
倉庫指標
- Star
- (392 star)
- PR 合併指標
- (PR 指標待抓取)
描述
Currently only integer RNG states are supported as per https://github.com/NVIDIA/Fuser/blob/e3a0b3ef528a86012d0a5268f3065b9203ed2784/csrc/python_frontend/python_bindings.cpp#L2426-L2473.
I'd like to feature request the support of 0-dim Tensor as rng_seed and rng_offset as well.
The dtype would be int64 and the device would be either cpu and cuda.
On top of that, I'd like the function to return a Tensor of integer or an integer that should be used to advance / increment the offset for the next function call.