Skip to content

GitLab

Commit 2a467090 authored by Syed Tousif Ahmed's avatar Syed Tousif Ahmed
Browse files

Rename IntList to IntArrayRef

parent 823b30cf
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 12 deletions
csrc/layer_norm_cuda.cpp
View file @ 2a467090
...@@ -5,7 +5,7 @@ ...@@ -5,7 +5,7 @@
namespace { namespace {
void compute_n1_n2( void compute_n1_n2(
at::Tensor input, at::Tensor input,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
int& n1, int& n1,
int& n2) int& n2)
{ {
...@@ -22,7 +22,7 @@ void compute_n1_n2( ...@@ -22,7 +22,7 @@ void compute_n1_n2(
} }
void check_args( void check_args(
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
at::Tensor gamma, at::Tensor gamma,
at::Tensor beta at::Tensor beta
) )
...@@ -33,7 +33,7 @@ void check_args( ...@@ -33,7 +33,7 @@ void check_args(
void check_args( void check_args(
at::Tensor input, at::Tensor input,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
int& n1, int& n1,
int& n2 int& n2
) )
...@@ -69,7 +69,7 @@ void check_args( ...@@ -69,7 +69,7 @@ void check_args(
void check_args( void check_args(
at::Tensor input, at::Tensor input,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
at::Tensor gamma, at::Tensor gamma,
at::Tensor beta, at::Tensor beta,
int& n1, int& n1,
...@@ -88,7 +88,7 @@ void cuda_layer_norm( ...@@ -88,7 +88,7 @@ void cuda_layer_norm(
at::Tensor* input, at::Tensor* input,
int n1, int n1,
int n2, int n2,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
at::Tensor* gamma, at::Tensor* gamma,
at::Tensor* beta, at::Tensor* beta,
double epsilon); double epsilon);
...@@ -99,7 +99,7 @@ void cuda_layer_norm( ...@@ -99,7 +99,7 @@ void cuda_layer_norm(
std::vector<at::Tensor> layer_norm( std::vector<at::Tensor> layer_norm(
at::Tensor input, at::Tensor input,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
double epsilon) { double epsilon) {
CHECK_INPUT(input); CHECK_INPUT(input);
int n1,n2; int n1,n2;
...@@ -113,7 +113,7 @@ std::vector<at::Tensor> layer_norm( ...@@ -113,7 +113,7 @@ std::vector<at::Tensor> layer_norm(
} }
std::vector<at::Tensor> layer_norm_affine( std::vector<at::Tensor> layer_norm_affine(
at::Tensor input, at::Tensor input,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
at::Tensor gamma, at::Tensor gamma,
at::Tensor beta, at::Tensor beta,
double epsilon) { double epsilon) {
...@@ -137,7 +137,7 @@ void cuda_layer_norm_gradient( ...@@ -137,7 +137,7 @@ void cuda_layer_norm_gradient(
at::Tensor* input, at::Tensor* input,
int n1, int n1,
int n2, int n2,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
at::Tensor* gamma, at::Tensor* gamma,
at::Tensor* beta, at::Tensor* beta,
double epsilon, double epsilon,
...@@ -151,7 +151,7 @@ at::Tensor layer_norm_gradient( ...@@ -151,7 +151,7 @@ at::Tensor layer_norm_gradient(
at::Tensor mean, at::Tensor mean,
at::Tensor invvar, at::Tensor invvar,
at::Tensor input, at::Tensor input,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
double epsilon) { double epsilon) {
CHECK_INPUT(dout); CHECK_INPUT(dout);
CHECK_INPUT(mean); CHECK_INPUT(mean);
...@@ -170,7 +170,7 @@ std::vector<at::Tensor> layer_norm_gradient_affine( ...@@ -170,7 +170,7 @@ std::vector<at::Tensor> layer_norm_gradient_affine(
at::Tensor mean, at::Tensor mean,
at::Tensor invvar, at::Tensor invvar,
at::Tensor input, at::Tensor input,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
at::Tensor gamma, at::Tensor gamma,
at::Tensor beta, at::Tensor beta,
double epsilon) { double epsilon) {
......
csrc/layer_norm_cuda_kernel.cu
View file @ 2a467090
...@@ -672,7 +672,7 @@ void cuda_layer_norm( ...@@ -672,7 +672,7 @@ void cuda_layer_norm(
at::Tensor* input, at::Tensor* input,
int n1, int n1,
int n2, int n2,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
at::Tensor* gamma, at::Tensor* gamma,
at::Tensor* beta, at::Tensor* beta,
double epsilon) double epsilon)
...@@ -767,7 +767,7 @@ void cuda_layer_norm_gradient( ...@@ -767,7 +767,7 @@ void cuda_layer_norm_gradient(
at::Tensor* input, at::Tensor* input,
int n1, int n1,
int n2, int n2,
at::IntList normalized_shape, at::IntArrayRef normalized_shape,
at::Tensor* gamma, at::Tensor* gamma,
at::Tensor* beta, at::Tensor* beta,
double epsilon, double epsilon,
......
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