wehub-resource-sync
77 lines
2.0 KiB
C++
77 lines
2.0 KiB
C++
// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
|
|
//
|
|
// 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.
|
|
#pragma once
|
|
#include <cuda_fp16.h>
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
namespace phi {
|
|
namespace fusion {
|
|
namespace cutlass_internal {
|
|
|
|
typedef enum {
|
|
fp32,
|
|
fp16,
|
|
bf16,
|
|
} Conv2dDataType;
|
|
|
|
typedef struct {
|
|
const void *input;
|
|
const void *weight;
|
|
const void *bias;
|
|
const void *residual;
|
|
void *output;
|
|
int batch;
|
|
int ic;
|
|
int ih;
|
|
int iw;
|
|
int kh;
|
|
int kw;
|
|
int oc;
|
|
int pad_h0;
|
|
int pad_h1;
|
|
int pad_w0;
|
|
int pad_w1;
|
|
int stride_h;
|
|
int stride_w;
|
|
int dilation_h;
|
|
int dilation_w;
|
|
int oh;
|
|
int ow;
|
|
int groups;
|
|
// const GPUContext *ctx;
|
|
cudaStream_t stream;
|
|
float alpha; // for leaky_relu use
|
|
int sm_version = 75;
|
|
Conv2dDataType data_type;
|
|
void *workspace = nullptr;
|
|
} ConvAllParams;
|
|
|
|
// Below functions are provided by cutlass, they are called by phi.
|
|
extern "C" bool Conv2dBiasAddRelu(ConvAllParams params);
|
|
extern "C" bool Conv2dBiasRelu(ConvAllParams params);
|
|
extern "C" bool Conv2dBiasLeakyRelu(ConvAllParams params);
|
|
extern "C" bool Conv2dBiasSilu(ConvAllParams params);
|
|
extern "C" bool Conv2dBias(ConvAllParams params);
|
|
extern "C" bool Conv2dBiasSigmoid(ConvAllParams params);
|
|
|
|
extern "C" bool Conv2dDepthwiseBias(ConvAllParams params);
|
|
extern "C" bool Conv2dDepthwiseBiasRelu(ConvAllParams params);
|
|
extern "C" bool Conv2dDepthwiseBiasSigmoid(ConvAllParams params);
|
|
extern "C" bool Conv2dDepthwiseBiasSilu(ConvAllParams params);
|
|
|
|
} // namespace cutlass_internal
|
|
} // namespace fusion
|
|
} // namespace phi
|