Move userbuffer to PyTorch - #162

Merged

ksivaman merged 7 commits into

NVIDIA:mainfrom

ksivaman:userbuf_refactor

Apr 21, 2023

Commits on Apr 20, 2023

Initial refactor; linker error
ksivaman
committed
fbbc325
Fix linking issue and make mpi conditional
ksivaman
committed
b930882
Fix TF/JAX build
ksivaman
committed
bfe3a8e
Use max SMs at the last RS chunk in pipelined overlap
ksivaman
and
erhoo82
committed
5103475
lint
ksivaman
committed
b46e291
Make userbuffers support opt-in
timmoon10
committed
d9f941b
Lint
timmoon10
committed
b5c6ba6