ATen
ATen copied to clipboard
Tensor cuda stream information
we need the cuda stream information for tensors to be used in c2isl. The only way to do this right now is via TH state.
we need the cuda stream information for tensors to be used in c2isl. The only way to do this right now is via TH state.