Sending the correct offset to the file.

#14: Refactor GpuFilteredRDD and FilteredChunkItrator Task-Url: http://github.com/meisam/spark/issues/issue/14
meisam · Oct 30, 2014 · dbee6e8 · dbee6e8
1 parent 375baea
commit dbee6e8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/core/src/main/scala/org/apache/spark/rdd/GpuPartition.scala b/core/src/main/scala/org/apache/spark/rdd/GpuPartition.scala
@@ -742,7 +742,8 @@ class GpuPartition[T <: Product : ClassTag]
 
   private def deviceToHostCopy[V: ClassTag](src: cl_mem, dest: Pointer, elementCount: Long, offset: Long = 0): Unit = {
     val length = elementCount * baseSize[V]
-    clEnqueueReadBuffer(context.getOpenCLQueue, src, CL_TRUE, offset, length, dest, 0, null,
+    val offsetInBytes = offset * baseSize[V]
+    clEnqueueReadBuffer(context.getOpenCLQueue, src, CL_TRUE, offsetInBytes, length, dest, 0, null,
       null)
   }