From 000334ea2a717a2b5bf2a61f021aad85bb79d9cb Mon Sep 17 00:00:00 2001
From: raver119 <raver119@gmail.com>
Date: Sat, 17 Aug 2019 17:59:14 +0300
Subject: [PATCH] affinity fix for tensorflow conversion

Signed-off-by: raver119 <raver119@gmail.com>
---
 .../org/nd4j/tensorflow/conversion/TensorflowConversion.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java b/nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java
index c5ae88198..b47cd30d1 100644
--- a/nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java
+++ b/nd4j/nd4j-tensorflow/src/main/java/org/nd4j/tensorflow/conversion/TensorflowConversion.java
@@ -296,7 +296,7 @@ public class TensorflowConversion {
      * @return
      */
     public static String defaultDeviceForThread() {
-        Integer deviceForThread = Nd4j.getAffinityManager().getDeviceForThread(Thread.currentThread());
+        Integer deviceForThread = Nd4j.getAffinityManager().getDeviceForCurrentThread();
         String deviceName = null;
         //gpu
         if(Nd4j.getBackend().getClass().getName().contains("JCublasBackend")) {