Skip to content

Commit b1c7cad

Browse files
committed
Merge pull request #26 from kiszk/CUDACodeGenerator
Add experimental version of GPU code generation
2 parents 8b65b12 + fabc292 commit b1c7cad

File tree

4 files changed

+690
-8
lines changed

4 files changed

+690
-8
lines changed

core/src/main/scala/org/apache/spark/SparkEnv.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@ class SparkEnv (
8484
val actorSystem: ActorSystem = _actorSystem
8585

8686
val isGPUEnabled = (cudaManager != null)
87+
val isGPUCodeGenEnabled = isGPUEnabled && conf.getBoolean("spark.gpu.codegen", false)
8788

8889
private[spark] var isStopped = false
8990
private val pythonWorkers = mutable.HashMap[(String, Map[String, String]), PythonWorkerFactory]()

0 commit comments

Comments
 (0)