do not relocate stuff on java side (#444)

Signed-off-by: raver119@gmail.com <raver119@gmail.com>
master
raver119 2020-05-09 12:56:45 +03:00 committed by GitHub
parent 2849299bfb
commit 10d6c3b024
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 1362 additions and 1382 deletions

View File

@ -106,7 +106,7 @@ public class CudaOpContext extends BaseOpContext implements OpContext, Deallocat
@Override
public void setInputArray(int index, @NonNull INDArray array) {
val ctx = AtomicAllocator.getInstance().getFlowController().prepareAction(null, array);
//val ctx = AtomicAllocator.getInstance().getFlowController().prepareAction(null, array);
nativeOps.setGraphContextInputBuffer(context, index, array.isEmpty() ? null : ((BaseCudaDataBuffer) array.data()).getOpaqueDataBuffer(), array.shapeInfoDataBuffer().addressPointer(), AtomicAllocator.getInstance().getPointer(array.shapeInfoDataBuffer()));
super.setInputArray(index, array);
@ -114,7 +114,7 @@ public class CudaOpContext extends BaseOpContext implements OpContext, Deallocat
@Override
public void setOutputArray(int index, @NonNull INDArray array) {
val ctx = AtomicAllocator.getInstance().getFlowController().prepareAction(array, null);
//val ctx = AtomicAllocator.getInstance().getFlowController().prepareAction(array, null);
nativeOps.setGraphContextOutputBuffer(context, index, array.isEmpty() ? null : ((BaseCudaDataBuffer) array.data()).getOpaqueDataBuffer(), array.shapeInfoDataBuffer().addressPointer(), AtomicAllocator.getInstance().getPointer(array.shapeInfoDataBuffer()));
super.setOutputArray(index, array);