pytorch/tools/codegen/dest
Edward Yang f3f7e92be5 Manually call lazyInitCUDA in structured CUDA calls (#61882)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61882

If you directly call the native implementation that bypasses the
initialization, which is bad!  This probably slows things down a little
though...

Fixes problem uncovered by #61642

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

Test Plan: Imported from OSS

Reviewed By: bhosmer

Differential Revision: D29783856

Pulled By: ezyang

fbshipit-source-id: 16857569a049e09c6ebd96ef04b0025403b254af
2021-07-22 07:50:05 -07:00
..
__init__.py add a boxed CPU fallback kernel (#58065) 2021-06-25 16:26:50 -07:00
native_functions.py move all external kernels into a class for better compiler error messages (#59839) 2021-07-08 15:31:02 -07:00
register_dispatch_key.py Manually call lazyInitCUDA in structured CUDA calls (#61882) 2021-07-22 07:50:05 -07:00