fix accident diff from future refactoring
@@ -172,7 +172,7 @@ namespace NKernel {
int tid = threadIdx.x;
result += blockIdx.x;
- TPointwisePartOffsetsHelper helper(foldCount);
+ TPartOffsetsHelper helper(foldCount);
for (int i = blockIdx.x * BLOCK_SIZE; i < binFeatureCount; i += BLOCK_SIZE * gridDim.x) {
if (i + tid >= binFeatureCount) {
@@ -266,7 +266,7 @@ namespace NKernel {
class TDirectHistLoader {
public:
__forceinline__ __device__ TDirectHistLoader(const float* binSums,
- TPointwisePartOffsetsHelper& helper,
+ TPartOffsetsHelper& helper,
int binFeatureId,
int /* leaf count*/,
int binFeatureCount)
@@ -285,7 +285,7 @@ namespace NKernel {
}
private:
const float* BinSums;
- TPointwisePartOffsetsHelper& Helper;
+ TPartOffsetsHelper& Helper;
int BinFeatureCount;
};
@@ -293,7 +293,7 @@ namespace NKernel {
class TGatheredByLeavesHistLoader {
public:
__forceinline__ __device__ TGatheredByLeavesHistLoader(const float* binSums,
- TPointwisePartOffsetsHelper&,
+ TPartOffsetsHelper&,
int binFeatureId,
int leafCount,
int /*binFeatureCount*/)
@@ -336,7 +336,7 @@ namespace NKernel {
int tid = threadIdx.x;
result += blockIdx.x;
- TPointwisePartOffsetsHelper helper(1);
+ TPartOffsetsHelper helper(1);
for (int i = blockIdx.x * BLOCK_SIZE; i < binFeatureCount; i += BLOCK_SIZE * gridDim.x) {
if (i + tid >= binFeatureCount) {
