 |
|
|
|
.. (parent) |
 |
d |
rwxr-xr-x |
69 |
no_python_abi_suffix_test
|
 |
d |
rwxr-xr-x |
90 |
open_registration_extension
|
 |
d |
rwxr-xr-x |
55 |
python_agnostic_extension
|
 |
d |
rwxr-xr-x |
26 |
self_compiler_include_dirs_test
|
 |
d |
rwxr-xr-x |
32 |
torch_test_cpp_extension
|
 |
- |
rw-r--r-- |
3,747 |
cpp_c10d_extension.cpp
|
 |
- |
rw-r--r-- |
3,914 |
cpp_c10d_extension.hpp
|
 |
- |
rw-r--r-- |
1,360 |
cpp_frontend_extension.cpp
|
 |
- |
rw-r--r-- |
422 |
cublas_extension.cpp
|
 |
- |
rw-r--r-- |
748 |
cuda_dlink_extension.cpp
|
 |
- |
rw-r--r-- |
139 |
cuda_dlink_extension_add.cu
|
 |
- |
rw-r--r-- |
127 |
cuda_dlink_extension_add.cuh
|
 |
- |
rw-r--r-- |
608 |
cuda_dlink_extension_kernel.cu
|
 |
- |
rw-r--r-- |
754 |
cuda_extension.cpp
|
 |
- |
rw-r--r-- |
1,158 |
cuda_extension.cu
|
 |
- |
rw-r--r-- |
779 |
cuda_extension_kernel.cu
|
 |
- |
rw-r--r-- |
780 |
cuda_extension_kernel2.cu
|
 |
- |
rw-r--r-- |
3,381 |
cudnn_extension.cpp
|
 |
- |
rw-r--r-- |
424 |
cusolver_extension.cpp
|
 |
- |
rw-r--r-- |
174 |
dangling_impl_extension.cpp
|
 |
- |
rw-r--r-- |
320 |
doubler.h
|
 |
- |
rw-r--r-- |
1,987 |
extension.cpp
|
 |
- |
rw-r--r-- |
531 |
identity.cpp
|
 |
- |
rw-r--r-- |
460 |
jit_extension.cpp
|
 |
- |
rw-r--r-- |
118 |
jit_extension2.cpp
|
 |
- |
rw-r--r-- |
4,339 |
maia_extension.cpp
|
 |
- |
rw-r--r-- |
1,968 |
mps_extension.mm
|
 |
- |
rw-r--r-- |
6,305 |
mtia_extension.cpp
|
 |
- |
rw-r--r-- |
31,111 |
open_registration_extension.cpp
|
 |
- |
rw-r--r-- |
2,715 |
rng_extension.cpp
|
 |
- |
rw-r--r-- |
3,427 |
setup.py
|
 |
- |
rw-r--r-- |
328 |
torch_library.cu
|