search

awslabs / raf

Apache License 2.0
140 stars 20 forks source link

[Op] Fix reduce_scatter infertype #121

Closed hgt312 closed 2 years ago

hgt312 commented 2 years ago

@comaniac

hgt312 commented 2 years ago

I just merged this after CI-lint passed, as raf only use 2 gpus for test, the change will not be tested