one more tiny cuda fux
Signed-off-by: raver119 <raver119@gmail.com>
This commit is contained in:
parent
82c9dc5743
commit
5b8ea3e830
@ -906,7 +906,7 @@ void clipByNormBP(nd4j::LaunchContext* context, const NDArray& input, const NDAr
|
||||
linearBuffers = shape::elementWiseStride(inputShape) == shape::elementWiseStride(outputShape) && shape::elementWiseStride(inputShape) == 1;
|
||||
}
|
||||
__syncthreads();
|
||||
const auto tid = blockIdx.x * gridDim.x + threadIdx.x;
|
||||
const auto tid = blockIdx.x * blockDim.x + threadIdx.x;
|
||||
const auto step = gridDim.x * blockDim.x;
|
||||
|
||||
for (Nd4jLong e = tid; e < length; e += step) {
|
||||
|
Loading…
x
Reference in New Issue
Block a user