hl_cuda_cudnn_stub.h 9.6 KB
Newer Older
Z
zhangjinchao01 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/* Copyright (c) 2016 Baidu, Inc. All Rights Reserve.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */

#ifndef HL_CUDA_CUDNN_STUB_H_
#define HL_CUDA_CUDNN_STUB_H_

#include "hl_cuda_cudnn.h"

20
inline int hl_get_cudnn_lib_version() { return 0; }
Z
zhangjinchao01 已提交
21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67

inline void hl_create_tensor_descriptor(hl_tensor_descriptor* image_desc) {}

inline void hl_tensor_reshape(hl_tensor_descriptor image_desc,
                              int batch_size,
                              int feature_maps,
                              int height,
                              int width) {}

inline void hl_tensor_reshape(hl_tensor_descriptor image_desc,
                              int batch_size,
                              int feature_maps,
                              int height,
                              int width,
                              int nStride,
                              int cStride,
                              int hStride,
                              int wStride) {}

inline void hl_destroy_tensor_descriptor(hl_tensor_descriptor image_desc) {}

inline void hl_create_pooling_descriptor(hl_pooling_descriptor* pooling_desc,
                                         hl_pooling_mode_t mode,
                                         int height,
                                         int width,
                                         int height_padding,
                                         int width_padding,
                                         int stride_height,
                                         int stride_width) {}

inline void hl_destroy_pooling_descriptor(hl_pooling_descriptor pooling_desc) {}

inline void hl_pooling_forward(hl_tensor_descriptor input,
                               real* input_image,
                               hl_tensor_descriptor output,
                               real* output_image,
                               hl_pooling_descriptor pooling) {}

inline void hl_pooling_backward(hl_tensor_descriptor input,
                                real* input_image,
                                real* input_image_grad,
                                hl_tensor_descriptor output,
                                real* output_image,
                                real* output_image_grad,
                                hl_pooling_descriptor pooling) {}

inline void hl_create_filter_descriptor(hl_filter_descriptor* filter,
68 69 70 71
                                        int input_feature_maps,
                                        int output_feature_maps,
                                        int height,
                                        int width) {}
Z
zhangjinchao01 已提交
72 73 74 75

inline void hl_destroy_filter_descriptor(hl_filter_descriptor filter) {}

inline void hl_create_convolution_descriptor(hl_convolution_descriptor* conv,
76 77 78 79 80 81
                                             hl_tensor_descriptor image,
                                             hl_filter_descriptor filter,
                                             int padding_height,
                                             int padding_width,
                                             int stride_height,
                                             int stride_width) {}
Z
zhangjinchao01 已提交
82 83

inline void hl_reset_convolution_descriptor(hl_convolution_descriptor conv,
84 85 86 87 88 89
                                            hl_tensor_descriptor image,
                                            hl_filter_descriptor filter,
                                            int padding_height,
                                            int padding_width,
                                            int stride_height,
                                            int stride_width) {}
Z
zhangjinchao01 已提交
90 91 92 93

inline void hl_destroy_convolution_descriptor(hl_convolution_descriptor conv) {}

inline void hl_conv_workspace(hl_tensor_descriptor input,
94 95 96 97 98 99 100 101 102
                              hl_tensor_descriptor output,
                              hl_filter_descriptor filter,
                              hl_convolution_descriptor conv,
                              int* convFwdAlgo,
                              size_t* fwdLimitBytes,
                              int* convBwdDataAlgo,
                              size_t* bwdDataLimitBytes,
                              int* convBwdFilterAlgo,
                              size_t* bwdFilterLimitBytes) {}
Z
zhangjinchao01 已提交
103 104 105 106 107 108 109 110 111 112 113 114 115

inline void hl_convolution_forward(hl_tensor_descriptor input,
                                   real* input_data,
                                   hl_tensor_descriptor output,
                                   real* output_data,
                                   hl_filter_descriptor filter,
                                   real* filter_data,
                                   hl_convolution_descriptor conv,
                                   void* gpuWorkSpace,
                                   size_t sizeInBytes,
                                   int convFwdAlgo) {}

inline void hl_convolution_forward_add_bias(hl_tensor_descriptor bias,
116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140
                                            real* bias_data,
                                            hl_tensor_descriptor output,
                                            real* output_data) {}

inline void hl_convolution_backward_filter(hl_tensor_descriptor input,
                                           real* input_data,
                                           hl_tensor_descriptor output,
                                           real* output_grad_data,
                                           hl_filter_descriptor filter,
                                           real* filter_grad_data,
                                           hl_convolution_descriptor conv,
                                           void* gpuWorkSpace,
                                           size_t sizeInBytes,
                                           int convBwdFilterAlgo) {}

inline void hl_convolution_backward_data(hl_tensor_descriptor input,
                                         real* input_data_grad,
                                         hl_tensor_descriptor output,
                                         real* output_grad_data,
                                         hl_filter_descriptor filter,
                                         real* filter_data,
                                         hl_convolution_descriptor conv,
                                         void* gpuWorkSpace,
                                         size_t sizeInBytes,
                                         int convBwdDataAlgo) {}
Z
zhangjinchao01 已提交
141 142

inline void hl_convolution_backward_bias(hl_tensor_descriptor bias,
143 144 145
                                         real* bias_grad_data,
                                         hl_tensor_descriptor output,
                                         real* output_grad_data) {}
Z
zhangjinchao01 已提交
146

147 148
inline void hl_softmax_forward(real* input,
                               real* output,
Z
zhangjinchao01 已提交
149 150 151
                               int height,
                               int width) {}

152 153 154 155 156
inline void hl_softmax_backward(real* output_value,
                                real* output_grad,
                                int height,
                                int width) {}

Z
zhangjinchao01 已提交
157
inline void hl_batch_norm_forward_training(hl_tensor_descriptor inputDesc,
158
                                           real* input,
Z
zhangjinchao01 已提交
159
                                           hl_tensor_descriptor outputDesc,
160
                                           real* output,
Z
zhangjinchao01 已提交
161
                                           hl_tensor_descriptor bnParamDesc,
162 163
                                           real* scale,
                                           real* bias,
Z
zhangjinchao01 已提交
164
                                           double factor,
165 166
                                           real* runningMean,
                                           real* runningInvVar,
Z
zhangjinchao01 已提交
167
                                           double epsilon,
168 169
                                           real* savedMean,
                                           real* savedVar) {}
Z
zhangjinchao01 已提交
170 171

inline void hl_batch_norm_forward_inference(hl_tensor_descriptor inputDesc,
172
                                            real* input,
Z
zhangjinchao01 已提交
173
                                            hl_tensor_descriptor outputDesc,
174
                                            real* output,
Z
zhangjinchao01 已提交
175
                                            hl_tensor_descriptor bnParamDesc,
176 177 178 179
                                            real* scale,
                                            real* bias,
                                            real* estimatedMean,
                                            real* estimatedVar,
Z
zhangjinchao01 已提交
180 181 182
                                            double epsilon) {}

inline void hl_batch_norm_backward(hl_tensor_descriptor inputDesc,
183
                                   real* input,
Z
zhangjinchao01 已提交
184
                                   hl_tensor_descriptor outGradDesc,
185
                                   real* outGrad,
Z
zhangjinchao01 已提交
186
                                   hl_tensor_descriptor inGradDesc,
187
                                   real* inGrad,
Z
zhangjinchao01 已提交
188
                                   hl_tensor_descriptor dBnParamDesc,
189 190 191
                                   real* scale,
                                   real* scaleGrad,
                                   real* biasGrad,
Z
zhangjinchao01 已提交
192
                                   double epsilon,
193 194
                                   real* savedMean,
                                   real* savedInvVar) {}
Z
zhangjinchao01 已提交
195 196

#endif  // HL_CUDA_CUDNN_STUB_H_