AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1023 from emankov/master

Browse Source 
[HIPIFY][cuDNN] Add partial cudnnRNNBiasMode_t support
This commit is contained in:
Evgeny Mankov
2019-04-16 11:03:22 +03:00
committed by GitHub
parent 8309632e2d 1b36987c5f
commit ea389cd7b8
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/markdown/CUDNN_API_supported_by_HIP.md
+7 -7
View File
@@ -2,8 +2,8 @@
## **1. CUDNN Data types**
| **type** | **CUDA** | **HIP** |
|-------------:|---------------------------------------------------------------|------------------------------------------------------------|
| **type** | **CUDA** | **HIP** |**HIP value** (if differs) |
|-------------:|---------------------------------------------------------------|------------------------------------------------------------|---------------------------|
| define |`CUDNN_VERSION` |`HIPDNN_VERSION` |
| struct |`cudnnContext` | |
| struct* |`cudnnHandle_t` |`hipdnnHandle_t` |
@@ -174,11 +174,11 @@
| 1 |*`CUDNN_RNN_TANH`* |*`HIPDNN_RNN_TANH`* |
| 2 |*`CUDNN_LSTM`* |*`HIPDNN_LSTM`* |
| 3 |*`CUDNN_GRU`* |*`HIPDNN_GRU`* |
| enum |***`cudnnRNNBiasMode_t`*** | |
| 0 |*`CUDNN_RNN_NO_BIAS`* | |
| 1 |*`CUDNN_RNN_SINGLE_INP_BIAS`* | |
| 2 |*`CUDNN_RNN_DOUBLE_BIAS`* | |
| 3 |*`CUDNN_RNN_SINGLE_REC_BIAS`* | |
| enum |***`cudnnRNNBiasMode_t`*** |***`hipdnnRNNBiasMode_t`*** |
| 0 |*`CUDNN_RNN_NO_BIAS`* |*`HIPDNN_RNN_NO_BIAS`* |
| 1 |*`CUDNN_RNN_SINGLE_INP_BIAS`* |*`HIPDNN_RNN_WITH_BIAS`* |
| 2 |*`CUDNN_RNN_DOUBLE_BIAS`* |*`HIPDNN_RNN_WITH_BIAS`* | 1 |
| 3 |*`CUDNN_RNN_SINGLE_REC_BIAS`* |*`HIPDNN_RNN_WITH_BIAS`* | 1 |
| enum |***`cudnnDirectionMode_t`*** |***`hipdnnDirectionMode_t`*** |
| 0 |*`CUDNN_UNIDIRECTIONAL`* |*`HIPDNN_UNIDIRECTIONAL`* |
| 1 |*`CUDNN_BIDIRECTIONAL`* |*`HIPDNN_BIDIRECTIONAL`* |
hipify-clang/src/CUDA2HIP_DNN_API_types.cpp
+5 -5
View File
@@ -151,11 +151,11 @@ const std::map<llvm::StringRef, hipCounter> CUDA_DNN_TYPE_NAME_MAP{
{"CUDNN_RNN_TANH", {"HIPDNN_RNN_TANH", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 1
{"CUDNN_LSTM", {"HIPDNN_LSTM", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 2
{"CUDNN_GRU", {"HIPDNN_GRU", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 3
{"cudnnRNNBiasMode_t", {"hipdnnRNNBiasMode_t", "", CONV_TYPE, API_DNN, HIP_UNSUPPORTED}},
{"CUDNN_RNN_NO_BIAS", {"HIPDNN_RNN_NO_BIAS", "", CONV_NUMERIC_LITERAL, API_DNN, HIP_UNSUPPORTED}}, // 0
{"CUDNN_RNN_SINGLE_INP_BIAS", {"HIPDNN_RNN_SINGLE_INP_BIAS", "", CONV_NUMERIC_LITERAL, API_DNN, HIP_UNSUPPORTED}}, // 1
{"CUDNN_RNN_DOUBLE_BIAS", {"HIPDNN_RNN_DOUBLE_BIAS", "", CONV_NUMERIC_LITERAL, API_DNN, HIP_UNSUPPORTED}}, // 2
{"CUDNN_RNN_SINGLE_REC_BIAS", {"HIPDNN_RNN_SINGLE_REC_BIAS", "", CONV_NUMERIC_LITERAL, API_DNN, HIP_UNSUPPORTED}}, // 3
{"cudnnRNNBiasMode_t", {"hipdnnRNNBiasMode_t", "", CONV_TYPE, API_DNN}},
{"CUDNN_RNN_NO_BIAS", {"HIPDNN_RNN_NO_BIAS", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 0
{"CUDNN_RNN_SINGLE_INP_BIAS", {"HIPDNN_RNN_WITH_BIAS", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 1
{"CUDNN_RNN_DOUBLE_BIAS", {"HIPDNN_RNN_WITH_BIAS", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 1
{"CUDNN_RNN_SINGLE_REC_BIAS", {"HIPDNN_RNN_WITH_BIAS", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 1
{"cudnnOpTensorOp_t", {"hipdnnOpTensorOp_t", "", CONV_TYPE, API_DNN}},
{"CUDNN_OP_TENSOR_ADD", {"HIPDNN_OP_TENSOR_ADD", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 0
{"CUDNN_OP_TENSOR_MUL", {"HIPDNN_OP_TENSOR_MUL", "", CONV_NUMERIC_LITERAL, API_DNN}}, // 1