[jira] [Commented] (SPARK-3785) Support off-loading computations to a GPU

["Kazuaki Ishizaki (JIRA)" <ji...@apache.org>]

    [ https://issues.apache.org/jira/browse/SPARK-3785?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15080653#comment-15080653 ] 

Kazuaki Ishizaki commented on SPARK-3785:
-----------------------------------------

Let us reopen this thread :)

We are working for effectively and easily exploiting GPUs on Spark at  [http://github.com/kiszk/spark-gpu]. Our project page is [http://kiszk.github.io/spark-gpu/]. A design document is [here|https://docs.google.com/document/d/1bo1hbQ7ikdUA9LYtYh6kU_TwjFK2ebkHsH66QlmbYP8/edit?usp=sharing]

Our ideas for exploiting GPUs are
# adding a new format for a partition in an RDD, which is a column-based structure in an array format, in addition to the current Iterator\[T\] format with Seq\[T\]
# generating parallelized GPU native code to access data in the new format from a Spark application program by using an optimizer and code generator (this is similar to [Project Tungsten|https://databricks.com/blog/2015/04/28/project-tungsten-bringing-spark-closer-to-bare-metal.html]) and pre-compiled library

The motivation of idea 1 is to reduce the overhead of serializing/deserializing partition data for copy between CPU and GPU. The motivation of idea 2 is to avoid writing hardware-dependent code by application programmers. At first, we are working for idea A (For idea B, we need to write [CUDA|https://en.wikipedia.org/wiki/CUDA] code for now). 

This prototype achieved [3.15x performance improvement|https://github.com/kiszk/spark-gpu/wiki/Benchmark] of logistic regression ([SparkGPULR|https://github.com/kiszk/spark-gpu/blob/dev/examples/src/main/scala/org/apache/spark/examples/SparkGPULR.scala]) in examples on a 16-thread IvyBridge box with an NVIDIA K40 GPU card over that with no GPU card

You can download the pre-build binary for x86_64 and ppc64le from [here|https://github.com/kiszk/spark-gpu/wiki/Downloads]. You can run this on Amazon EC2 by [the procedure|https://github.com/kiszk/spark-gpu/wiki/How-to-run-%28local-or-AWS-EC2%29], too.


> Support off-loading computations to a GPU
> -----------------------------------------
>
>                 Key: SPARK-3785
>                 URL: https://issues.apache.org/jira/browse/SPARK-3785
>             Project: Spark
>          Issue Type: Brainstorming
>          Components: MLlib
>            Reporter: Thomas Darimont
>            Priority: Minor
>
> Are there any plans to adding support for off-loading computations to the GPU, e.g. via an open-cl binding? 
> http://www.jocl.org/
> https://code.google.com/p/javacl/
> http://lwjgl.org/wiki/index.php?title=OpenCL_in_LWJGL



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org