Nice summary: https://blog.csdn.net/HaoBBNuanMM/article/details/112243129
- https://github.com/pytorch/pytorch/blob/e686a442b45885aea4a1c38273a82cc3ca6c09fa/test/cpp_extensions/setup.py#L82
- https://github.com/pytorch/pytorch/blob/e686a442b45885aea4a1c38273a82cc3ca6c09fa/test/test_cpp_extensions_aot.py#L333
- https://github.com/pytorch/pytorch/blob/e686a442b45885aea4a1c38273a82cc3ca6c09fa/test/cpp_extensions/torch_library.cu#L5