parent
df84bc7255
commit
05d45ec050
|
@ -894,13 +894,13 @@ public class CudaExecutioner extends DefaultOpExecutioner {
|
|||
|
||||
//long dimensionPointer = AtomicAllocator.getInstance().getPointer(Nd4j.createBuffer(dimension), context);
|
||||
Pointer dimensionPointer = AtomicAllocator.getInstance()
|
||||
.getPointer(AtomicAllocator.getInstance().getConstantBuffer(dimension), context);
|
||||
.getHostPointer(AtomicAllocator.getInstance().getConstantBuffer(dimension));
|
||||
|
||||
nativeOps.execIndexReduce(xShapeInfoHostPointer, op.opNum(),
|
||||
null, (LongPointer) hostXShapeInfo, x, (LongPointer) xShapeInfo,
|
||||
extraArgs,
|
||||
null, (LongPointer) hostZShapeInfo, z, (LongPointer) zShapeInfo,
|
||||
null,
|
||||
dimensionPointer,
|
||||
(LongPointer) op.dimensions().shapeInfoDataBuffer().addressPointer(),
|
||||
AtomicAllocator.getInstance().getPointer(op.dimensions(), context),
|
||||
null);
|
||||
|
|
Loading…
Reference in New Issue