"...composable_kernel.git" did not exist on "6dfb92bbef33b4caea55f6b4ed7c449927ae771c"
Commit bb314592 authored by rocking's avatar rocking
Browse files

Remove epislon

parent 459f63a8
...@@ -83,7 +83,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator ...@@ -83,7 +83,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
const GammaDataType* p_gamma, const GammaDataType* p_gamma,
const BetaDataType* p_beta, const BetaDataType* p_beta,
OutDataType* p_output, OutDataType* p_output,
MeanSquareDataType epsilon,
const std::vector<index_t>& shape, const std::vector<index_t>& shape,
const std::vector<index_t>& stride_x, const std::vector<index_t>& stride_x,
const std::vector<index_t>& stride_mean, const std::vector<index_t>& stride_mean,
...@@ -98,7 +97,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator ...@@ -98,7 +97,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
p_gamma_(p_gamma), p_gamma_(p_gamma),
p_beta_(p_beta), p_beta_(p_beta),
p_output_(p_output), p_output_(p_output),
epsilon_(epsilon),
shape_(shape), shape_(shape),
stride_x_(stride_x), stride_x_(stride_x),
stride_mean_(stride_mean), stride_mean_(stride_mean),
...@@ -124,7 +122,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator ...@@ -124,7 +122,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
const GammaDataType* p_gamma_; const GammaDataType* p_gamma_;
const BetaDataType* p_beta_; const BetaDataType* p_beta_;
OutDataType* p_output_; OutDataType* p_output_;
MeanSquareDataType epsilon_;
std::vector<index_t> shape_; std::vector<index_t> shape_;
GridDesc_M0 x_grid_desc_m0_; GridDesc_M0 x_grid_desc_m0_;
GridDesc_M0 mean_grid_desc_m0_; GridDesc_M0 mean_grid_desc_m0_;
...@@ -179,6 +176,9 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator ...@@ -179,6 +176,9 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
if(pArg == nullptr) if(pArg == nullptr)
return false; return false;
if(pArg->shape_.size() != Dim)
return false;
if(pArg->shape_.back() % M0PerThread != 0) if(pArg->shape_.back() % M0PerThread != 0)
return false; return false;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment