Support setting core number from python side (intel-analytics#1509)

* support set core num * fix session_config * fix license * address comments
Le-Zheng · Jul 15, 2019 · 211b3a8 · 211b3a8
1 parent 7a64fd6
commit 211b3a8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/layers/utils/Reflection.scala b/layers/utils/Reflection.scala
@@ -88,6 +88,12 @@ object EngineRef {
   def getEngineType(): EngineType = {
     KerasUtils.invokeMethod(Engine, "getEngineType").asInstanceOf[EngineType]
   }
+
+  def setCoreNumber(num: Int): Unit = {
+    val field = Engine.getClass.getDeclaredField("physicalCoreNumber")
+    field.setAccessible(true)
+    field.setInt(Engine, num)
+  }
 }
 
 object SGDRef {