Skip to content

Commit a7d82b9

Browse files
committed
Merge commit '9851ef4979bad0c8618e586e711c1bfd8648fd52'
2 parents af7aea9 + 9851ef4 commit a7d82b9

File tree

2 files changed

+14
-14
lines changed

2 files changed

+14
-14
lines changed

torch/lib/THC/generic/THCTensorMathReduce.cu

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ THCTensor_(renorm)(THCState *state, THCTensor* self, THCTensor* src, real value,
7272
}
7373

7474
THC_API void
75-
THCTensor_(std)(THCState *state, THCTensor *self_, THCTensor *src, long dimension, int flag, int keepdim)
75+
THCTensor_(std)(THCState *state, THCTensor *self_, THCTensor *src, long dimension, int biased, int keepdim)
7676
{
7777
THCAssertSameGPU(THCTensor_(checkGPU)(state, 2, self_, src));
7878
THLongStorage *dim = THCTensor_(newSizeOf)(state, src);
@@ -84,9 +84,9 @@ THCTensor_(std)(THCState *state, THCTensor *self_, THCTensor *src, long dimensio
8484
src = THCTensor_(newContiguous)(state, src);
8585

8686
if (dimension == THCTensor_(nDimension)(state, src) - 1) {
87-
THCTensor_varInnermostDim<THCTensor, real, true>(state, self, src, flag);
87+
THCTensor_varInnermostDim<THCTensor, real, true>(state, self, src, biased);
8888
} else {
89-
THCTensor_varOuterDim<THCTensor, real, true>(state, self, src, dimension, flag);
89+
THCTensor_varOuterDim<THCTensor, real, true>(state, self, src, dimension, biased);
9090
}
9191

9292
THCTensor_(free)(state, src);
@@ -98,7 +98,7 @@ THCTensor_(std)(THCState *state, THCTensor *self_, THCTensor *src, long dimensio
9898
}
9999

100100
THC_API void
101-
THCTensor_(var)(THCState *state, THCTensor *self_, THCTensor *src, long dimension, int flag, int keepdim)
101+
THCTensor_(var)(THCState *state, THCTensor *self_, THCTensor *src, long dimension, int biased, int keepdim)
102102
{
103103
THCAssertSameGPU(THCTensor_(checkGPU)(state, 2, self_, src));
104104
THLongStorage *dim = THCTensor_(newSizeOf)(state, src);
@@ -110,9 +110,9 @@ THCTensor_(var)(THCState *state, THCTensor *self_, THCTensor *src, long dimensio
110110
src = THCTensor_(newContiguous)(state, src);
111111

112112
if (dimension == THCTensor_(nDimension)(state, src) - 1) {
113-
THCTensor_varInnermostDim<THCTensor, real, false>(state, self, src, flag);
113+
THCTensor_varInnermostDim<THCTensor, real, false>(state, self, src, biased);
114114
} else {
115-
THCTensor_varOuterDim<THCTensor, real, false>(state, self, src, dimension, flag);
115+
THCTensor_varOuterDim<THCTensor, real, false>(state, self, src, dimension, biased);
116116
}
117117

118118
THCTensor_(free)(state, src);
@@ -124,14 +124,14 @@ THCTensor_(var)(THCState *state, THCTensor *self_, THCTensor *src, long dimensio
124124
}
125125

126126
THC_API accreal
127-
THCTensor_(stdall)(THCState *state, THCTensor *self)
127+
THCTensor_(stdall)(THCState *state, THCTensor *self, int biased)
128128
{
129129
THCAssertSameGPU(THCTensor_(checkGPU)(state, 1, self));
130-
return THCNumerics<accreal>::sqrt((THCTensor_(varall)(state, self)));
130+
return THCNumerics<accreal>::sqrt((THCTensor_(varall)(state, self, biased)));
131131
}
132132

133133
THC_API accreal
134-
THCTensor_(varall)(THCState *state, THCTensor *self)
134+
THCTensor_(varall)(THCState *state, THCTensor *self, int biased)
135135
{
136136
THCAssertSameGPU(THCTensor_(checkGPU)(state, 1, self));
137137
accreal mean = THCTensor_(meanall)(state, self);
@@ -148,7 +148,7 @@ THCTensor_(varall)(THCState *state, THCTensor *self)
148148

149149
val = THCNumerics<accreal>::div(
150150
val,
151-
ScalarConvert<ptrdiff_t, accreal>::to(THCTensor_(nElement)(state, self) - 1)
151+
ScalarConvert<ptrdiff_t, accreal>::to(THCTensor_(nElement)(state, self) - (biased ? 0 : 1))
152152
);
153153

154154
THCudaCheck(cudaGetLastError());

torch/lib/THC/generic/THCTensorMathReduce.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,13 @@
55
#if defined(THC_REAL_IS_FLOAT) || defined(THC_REAL_IS_DOUBLE) || defined(THC_REAL_IS_HALF)
66

77
THC_API void THCTensor_(renorm)(THCState *state, THCTensor* self, THCTensor* src, real value, long dimension, real max_norm);
8-
THC_API void THCTensor_(std)(THCState *state, THCTensor *self, THCTensor *src, long dim, int flag, int keepdim);
8+
THC_API void THCTensor_(std)(THCState *state, THCTensor *self, THCTensor *src, long dim, int biased, int keepdim);
99
THC_API void THCTensor_(norm)(THCState *state, THCTensor* self, THCTensor* src, real value, long dimension, int keepdim);
10-
THC_API void THCTensor_(var)(THCState *state, THCTensor *self, THCTensor *src, long dim, int flag, int keepdim);
10+
THC_API void THCTensor_(var)(THCState *state, THCTensor *self, THCTensor *src, long dim, int biased, int keepdim);
1111

12-
THC_API accreal THCTensor_(stdall)(THCState *state, THCTensor *self);
12+
THC_API accreal THCTensor_(stdall)(THCState *state, THCTensor *self, int biased);
1313
THC_API accreal THCTensor_(normall)(THCState *state, THCTensor *self, real value);
14-
THC_API accreal THCTensor_(varall)(THCState *state, THCTensor *self);
14+
THC_API accreal THCTensor_(varall)(THCState *state, THCTensor *self, int biased);
1515

1616
#endif
1717

0 commit comments

Comments
 (0)