testing

Signed-off-by: minmingzhu <minming.zhu@intel.com>
oap-project · Aug 8, 2023 · f2e46c1 · f2e46c1
1 parent 61459f8
commit f2e46c1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/mllib-dal/src/main/java/com/intel/oneapi/dal/table/HomogenTableImpl.java b/mllib-dal/src/main/java/com/intel/oneapi/dal/table/HomogenTableImpl.java
@@ -77,8 +77,12 @@ public HomogenTableImpl(long rowCount,
                 this.cObject = fPtrInit(rowCount, colCount, dataPtr, dataLayout.ordinal(),
                         this.device.ordinal());
                 break;
+            case FLOAT64:
+                this.cObject = dPtrInit(rowCount, colCount, dataPtr, dataLayout.ordinal(),
+                        this.device.ordinal());
+                break;
             default:
-                System.err.println("oneapi only support float type to fix DAALL-6046");
+                System.err.println("oneapi only support float/double type to fix DAALL-6046");
                 System.exit(-1);
         }
     }

diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/OneDAL.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/OneDAL.scala
@@ -476,7 +476,7 @@ object OneDAL {
       }
       val labelsTable = new HomogenTable(numRows.toLong, 1, labelsArray,
         device)
-      val featuresTable = new HomogenTable(numRows.toLong, numCols.toLong, featuresAddress, Common.DataType.FLOAT32,
+      val featuresTable = new HomogenTable(numRows.toLong, numCols.toLong, featuresAddress, Common.DataType.FLOAT64,
         device)
 
       Iterator((featuresTable.getcObejct(), labelsTable.getcObejct()))
@@ -648,7 +648,7 @@ object OneDAL {
       Await.result(result, Duration.Inf)
       }
       val table = new HomogenTable(numRows.toLong, numCols.toLong, targetArrayAddress,
-        Common.DataType.FLOAT32, device)
+        Common.DataType.FLOAT64, device)
 
       Iterator(table.getcObejct())
     }.setName("coalescedTables").cache()