mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39277 This PR contains initial changes that makes PyTorch build with Ampere GPU, CUDA 11, and cuDNN 8. TF32 related features will not be included in this PR. Test Plan: Imported from OSS Differential Revision: D21832814 Pulled By: malfet fbshipit-source-id: 37f9c6827e0c26ae3e303580f666584230832d06 |
||
|---|---|---|
| .. | ||
| hip | ||
| math | ||
| threadpool | ||
| bench_utils.cc | ||
| bench_utils.h | ||
| cast.h | ||
| cast_test.cc | ||
| cblas.h | ||
| CMakeLists.txt | ||
| conversions.h | ||
| cpu_neon.h | ||
| cpuid.cc | ||
| cpuid.h | ||
| cpuid_test.cc | ||
| eigen_utils.h | ||
| fatal_signal_asan_no_sig_test.cc | ||
| filler.h | ||
| fixed_divisor.h | ||
| fixed_divisor_test.cc | ||
| GpuBitonicSort.cuh | ||
| GpuDefs.cuh | ||
| GpuScanUtils.cuh | ||
| map_utils.h | ||
| math-detail.h | ||
| math.h | ||
| math_cpu.cc | ||
| math_gpu.cu | ||
| math_gpu_test.cc | ||
| math_test.cc | ||
| murmur_hash3.cc | ||
| murmur_hash3.h | ||
| proto_convert.cc | ||
| proto_convert.h | ||
| proto_utils.cc | ||
| proto_utils.h | ||
| proto_utils_test.cc | ||
| proto_wrap.cc | ||
| proto_wrap.h | ||
| signal_handler.cc | ||
| signal_handler.h | ||
| simple_queue.h | ||
| simple_queue_test.cc | ||
| smart_tensor_printer.cc | ||
| smart_tensor_printer.h | ||
| smart_tensor_printer_test.cc | ||
| string_utils.cc | ||
| string_utils.h | ||
| zmq_helper.h | ||