Ashish 5ae3b44255 Added HIP top_k operator (#13747) ...

Summary:
This PR contains changes for:
1. Adding HIP top_k operator in Caffe2
2. Added HIP equivalent definitions of GPUDefs and GPUScanUtils
3. Removing the top_k operator test from ROCm test ignore list
4. Bug fixes in related code in THC/THCAsmUtils.cuh

Differential Revision: D12986451

Pulled By: bddppq

fbshipit-source-id: 6d5241fb674eaeb7cde42166426ac88043b83504

2018-11-08 20:14:53 -08:00

..

hip

Hipify caffe2/utils/math_gpu.cu (#13521)

2018-11-07 11:34:15 -08:00

threadpool

Using c10 namespace across caffe2. (#12714)

2018-10-17 12:57:19 -07:00

bench_utils.cc

wipe cache with writes (#12279)

2018-10-03 17:12:23 -07:00

bench_utils.h

build changes to make cpu unified build working. (#10504)

2018-08-15 17:22:36 -07:00

cast_test.cc

Update from facebook (#7855)

2018-05-29 11:38:02 -07:00

cast.h

Update from facebook (#7855)

2018-05-29 11:38:02 -07:00

cblas.h

Fix more MKL build issues

2017-08-25 14:01:01 -07:00

CMakeLists.txt

Move some files to c10/util (#12245)

2018-10-15 16:25:12 -07:00

conversions.h

Replace float16 with at::Half in caffe2 (#11676)

2018-09-20 18:55:17 -07:00

cpu_neon.h

[caffe2] Use both __ARM_NEON__ and __ARM_NEON macros (#6697)

2018-04-18 17:45:47 -04:00

cpuid_test.cc

Remove Apache headers from source.

2018-03-27 13:10:18 -07:00

cpuid.cc

Remove Apache headers from source.

2018-03-27 13:10:18 -07:00

cpuid.h

Remove Apache headers from source.

2018-03-27 13:10:18 -07:00

dummy.cpp

c10 build setup (#8264)

2018-06-08 12:11:17 -07:00

eigen_utils.h

Update Im2Col related to make preparation for group conv in NHWC order. (#10439)

2018-08-15 17:10:24 -07:00

fatal_signal_asan_no_sig_test.cc

Forward declarations of needed curand functions (#10911)

2018-09-05 16:56:26 -07:00

filler.h

Renaming size() to numel() - 2/2

2018-10-29 18:59:29 -07:00

fixed_divisor_test.cc

Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072)

2018-07-01 00:25:34 -07:00

fixed_divisor.h

Eanble python tests on ROCM (#9616)

2018-07-24 11:37:58 -07:00

GpuBitonicSort.cuh

Manually applying cudnn5 pull request.

2018-01-02 15:31:33 -08:00

GpuDefs.cuh

Added HIP top_k operator (#13747)

2018-11-08 20:14:53 -08:00

GpuScanUtils.cuh

Added HIP top_k operator (#13747)

2018-11-08 20:14:53 -08:00

map_utils.h

Remove Apache headers from source.

2018-03-27 13:10:18 -07:00

math_cpu.cc

3D Conv in NHWC layout (#12733)

2018-11-04 21:50:09 -08:00

math_gpu_test.cc

Renaming size() to numel() - 2/2

2018-10-29 18:59:29 -07:00

math_gpu.cu

Hipify caffe2/utils/math_gpu.cu (#13521)

2018-11-07 11:34:15 -08:00

math_test.cc

Renaming size() to numel() - 2/2

2018-10-29 18:59:29 -07:00

math_utils.cc

Optimize NCHW2NHWC on GPU (#12910)

2018-10-22 11:24:29 -07:00

math_utils.h

Optimize NCHW2NHWC on GPU (#12910)

2018-10-22 11:24:29 -07:00

math-detail.h

Remove Apache headers from source.

2018-03-27 13:10:18 -07:00

math.h

3D Conv in NHWC layout (#12733)

2018-11-04 21:50:09 -08:00

murmur_hash3.cc

Remove core and util warnings (#8239)

2018-06-07 09:10:33 -07:00

murmur_hash3.h

Fix race in FileStoreHandler

2017-02-03 09:59:45 -08:00

proto_convert.cc

New serialization format (#12384)

2018-10-16 16:36:58 -07:00

proto_convert.h

New serialization format (#12384)

2018-10-16 16:36:58 -07:00

proto_utils_test.cc

Rename cuda_gpu_id to device_id in DeviceOption (#12456)

2018-10-09 15:54:04 -07:00

proto_utils.cc

Add back HIP support to async net (#13400)

2018-10-31 17:52:36 -07:00

proto_utils.h

Add back HIP support to async net (#13400)

2018-10-31 17:52:36 -07:00

proto_wrap.cc

New Serialization Proto

2018-09-11 10:55:43 -07:00

proto_wrap.h

build changes to make cpu unified build working. (#10504)

2018-08-15 17:22:36 -07:00

signal_handler.cc

Using c10 namespace across caffe2. (#12714)

2018-10-17 12:57:19 -07:00

signal_handler.h

More changes for hidden visibility (#10692)

2018-08-21 13:39:57 -07:00

simple_queue_test.cc

Remove Apache headers from source.

2018-03-27 13:10:18 -07:00

simple_queue.h

Remove Apache headers from source.

2018-03-27 13:10:18 -07:00

smart_tensor_printer_test.cc

Kill more weird constructors on Tensor

2018-11-04 16:54:49 -08:00

smart_tensor_printer.cc

Renaming meta() to dtype() - 2/2 (#13334)

2018-10-30 18:24:30 -07:00

smart_tensor_printer.h

More changes for hidden visibility (#10692)

2018-08-21 13:39:57 -07:00

string_utils.cc

Update from Facebook (#8887)

2018-06-26 14:55:48 -07:00

string_utils.h

build changes to make cpu unified build working. (#10504)

2018-08-15 17:22:36 -07:00

thread_name.cc

Update from facebook (#8384)

2018-06-13 13:10:45 -07:00

thread_name.h

build changes to make cpu unified build working. (#10504)

2018-08-15 17:22:36 -07:00

thread_pool.h

Back out "Back out "[caffe2] Use custom CPU thread pool in async_scheduling"" (#12418)

2018-10-08 16:24:07 -07:00

zmq_helper.h

Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019)

2018-09-25 17:41:05 -07:00