Merge pull request #1115 from emankov/master

[HIPIFY][cuDNN] Add cudnnGetFilter4dDescriptor support
Этот коммит содержится в:
Evgeny Mankov
2019-05-16 16:39:34 +03:00
коммит произвёл GitHub
родитель ccfb764a59 aed2affda2
Коммит 390db35dfb
3 изменённых файлов: 15 добавлений и 2 удалений
+13
Просмотреть файл
@@ -176,6 +176,19 @@ int main() {
out_desc, CUDNN_TENSOR_NCHW, CUDNN_DATA_FLOAT,
out_n, out_c, out_h, out_w));
cudnnDataType_t *dataType = nullptr;
cudnnTensorFormat_t *tensorFormat = nullptr;
int *p_filt_k = nullptr;
int *p_filt_c = nullptr;
int *p_filt_h = nullptr;
int *p_filt_w = nullptr;
// CHECK: CUDNN_CALL(hipdnnGetFilter4dDescriptor(
CUDNN_CALL(cudnnGetFilter4dDescriptor(
filt_desc, dataType, tensorFormat,
p_filt_k, p_filt_c, p_filt_h, p_filt_w));
float *out_data;
// CHECK: CUDA_CALL(hipMalloc(
CUDA_CALL(cudaMalloc(