[incubator-mxnet] branch master updated: [MXNET-166] add unknown dtype to resolve segfaults when inputType is -1 and type … (#10428)

[li...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

liuyizhi pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new c3a6f19  [MXNET-166] add unknown dtype to resolve segfaults when inputType is -1 and type … (#10428)
c3a6f19 is described below

commit c3a6f19b55b2048e4326645979655597b6c39dbe
Author: Naveen Swamy <mn...@gmail.com>
AuthorDate: Thu Apr 5 20:20:21 2018 -0700

    [MXNET-166] add unknown dtype to resolve segfaults when inputType is -1 and type … (#10428)
    
    * add unknown dtype to resolve segfaults when inputType is -1 and type is unknown
    
    * change log info on MXNetHandler to debug
---
 scala-package/core/src/main/scala/ml/dmlc/mxnet/DType.scala             | 2 ++
 .../infer/src/main/scala/ml/dmlc/mxnet/infer/MXNetHandler.scala         | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/scala-package/core/src/main/scala/ml/dmlc/mxnet/DType.scala b/scala-package/core/src/main/scala/ml/dmlc/mxnet/DType.scala
index bfe757d..4ba101b 100644
--- a/scala-package/core/src/main/scala/ml/dmlc/mxnet/DType.scala
+++ b/scala-package/core/src/main/scala/ml/dmlc/mxnet/DType.scala
@@ -24,6 +24,7 @@ object DType extends Enumeration {
   val Float16 = Value(2, "float16")
   val UInt8 = Value(3, "uint8")
   val Int32 = Value(4, "int32")
+  val Unknown = Value(-1, "unknown")
   private[mxnet] def numOfBytes(dtype: DType): Int = {
     dtype match {
       case DType.UInt8 => 1
@@ -31,6 +32,7 @@ object DType extends Enumeration {
       case DType.Float16 => 2
       case DType.Float32 => 4
       case DType.Float64 => 8
+      case DType.Unknown => 0
     }
   }
 }
diff --git a/scala-package/infer/src/main/scala/ml/dmlc/mxnet/infer/MXNetHandler.scala b/scala-package/infer/src/main/scala/ml/dmlc/mxnet/infer/MXNetHandler.scala
index 47ee15c..a58e2c9 100644
--- a/scala-package/infer/src/main/scala/ml/dmlc/mxnet/infer/MXNetHandler.scala
+++ b/scala-package/infer/src/main/scala/ml/dmlc/mxnet/infer/MXNetHandler.scala
@@ -70,7 +70,7 @@ private[infer] class MXNetThreadPoolHandler(numThreads: Int = 1)
 
       val task = new Callable[T] {
         override def call(): T = {
-          logger.info("threadId: %s".format(Thread.currentThread().getId()))
+          logger.debug("threadId: %s".format(Thread.currentThread().getId()))
           f
         }
       }

-- 
To stop receiving notification emails like this one, please contact
liuyizhi@apache.org.