Shan19900305 7931eee5c5 Support torch.dtype as parameter in pybind11 cpp extension. (#126865) ...

Support torch.dtype as parameter in pybind11 cpp extension.
Example:
`
cpp_extension.my_ops(self, other, torch.dtype)
`

@ezyang @bdhirsh
Co-authored-by: Edward Z. Yang <ezyang@mit.edu>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/126865
Approved by: https://github.com/ezyang

2024-05-29 23:19:32 +00:00

..

no_python_abi_suffix_test

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)

2024-05-27 14:49:57 +00:00

self_compiler_include_dirs_test

…

torch_test_cpp_extension

…

cpp_c10d_extension.cpp

…

cpp_c10d_extension.hpp

…

cpp_frontend_extension.cpp

…

cublas_extension.cpp

…

cuda_dlink_extension_add.cu

…

cuda_dlink_extension_add.cuh

…

cuda_dlink_extension_kernel.cu

…

cuda_dlink_extension.cpp

…

cuda_extension_kernel2.cu

…

cuda_extension_kernel.cu

…

cuda_extension.cpp

…

cuda_extension.cu

…

cudnn_extension.cpp

…

cusolver_extension.cpp

…

dangling_impl_extension.cpp

…

doubler.h

…

extension.cpp

Support torch.dtype as parameter in pybind11 cpp extension. (#126865)

2024-05-29 23:19:32 +00:00

identity.cpp

…

jit_extension2.cpp

…

jit_extension.cpp

…

maia_extension.cpp

[codemod] c10::optional -> std::optional in caffe2/aten/src/ATen/DeviceGuard.h +117 (#126901)

2024-05-24 00:26:15 +00:00

mps_extension.mm

…

mtia_extension.cpp

Support generic stream/event on CUDA/HIP backend (#125757)

2024-05-10 13:34:09 +00:00

open_registration_extension.cpp

[codemod] c10:optional -> std::optional (#126135)

2024-05-14 19:35:51 +00:00

rng_extension.cpp

[codemod] c10:optional -> std::optional (#126135)

2024-05-14 19:35:51 +00:00

setup.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)

2024-05-27 14:49:57 +00:00

torch_library.cu

…