Jiawei Li

Results 28 comments of Jiawei Li

Hi @ezyang, please have to take a look at this pr, thank you.

> > What needs to be added is that we found the docker image for building x86, but we did not find the build environment for aarch64. > > I'm...

@ptrblck , sorry to @you again, can you help answer the above two questions? Thank you a lot.

From my point of view, compared to [DeviceGuardImplInterface](https://github.com/pytorch/pytorch/blob/main/c10/core/impl/DeviceGuardImplInterface.h#L57) , it is maybe better to implement this device-agnostic api on top of [AcceleratorHooksInterface](https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/detail/AcceleratorHooksInterface.h#L14), which should be positioned to provide a common...