pytorch/caffe2/utils
Xiang Gao b3fac8af6b Initial support for building on Ampere GPU, CUDA 11, cuDNN 8 (#39277)
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
2020-06-02 10:03:42 -07:00
..
hip
math [caffe2] fix Transpose2D calls in NHWC<->NCHW (#34625) 2020-03-16 15:18:44 -07:00
threadpool Do not define C10_IOS on Mac (#37283) 2020-04-25 13:52:46 -07:00
bench_utils.cc
bench_utils.h
cast.h
cast_test.cc
cblas.h
CMakeLists.txt Fix CMake errors on systems where {Q/X}NNPACK is not supported (#35607) 2020-04-24 12:37:23 -07:00
conversions.h [caffe2] use Clang identification macro in various places (#33574) 2020-02-20 15:16:11 -08:00
cpu_neon.h
cpuid.cc
cpuid.h [caffe2] Use cpuinfo in perfkernels to simplify build dependency (#36371) 2020-04-10 13:26:34 -07:00
cpuid_test.cc
eigen_utils.h
fatal_signal_asan_no_sig_test.cc
filler.h
fixed_divisor.h [caffe2] use Clang identification macro in various places (#33574) 2020-02-20 15:16:11 -08:00
fixed_divisor_test.cc
GpuBitonicSort.cuh
GpuDefs.cuh Initial support for building on Ampere GPU, CUDA 11, cuDNN 8 (#39277) 2020-06-02 10:03:42 -07:00
GpuScanUtils.cuh RIP CUDA <9.2: circleci, aten, and caffe2 (#36846) 2020-05-18 13:41:05 -07:00
map_utils.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
math-detail.h
math.h
math_cpu.cc Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
math_gpu.cu RIP CUDA <9.2: circleci, aten, and caffe2 (#36846) 2020-05-18 13:41:05 -07:00
math_gpu_test.cc
math_test.cc
murmur_hash3.cc
murmur_hash3.h
proto_convert.cc
proto_convert.h
proto_utils.cc Logging for ATen op subtype 2020-04-19 23:02:50 -07:00
proto_utils.h Logging for ATen op subtype 2020-04-19 23:02:50 -07:00
proto_utils_test.cc
proto_wrap.cc
proto_wrap.h
signal_handler.cc preprocessor cleanup (#33957) 2020-03-02 13:37:19 -08:00
signal_handler.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
simple_queue.h
simple_queue_test.cc
smart_tensor_printer.cc preprocessor cleanup (#33957) 2020-03-02 13:37:19 -08:00
smart_tensor_printer.h
smart_tensor_printer_test.cc
string_utils.cc
string_utils.h Fix out-of-boundary access in caffe2::StartsWith (#36672) 2020-04-15 20:40:59 -07:00
zmq_helper.h Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00