[
https://issues.apache.org/jira/browse/SYSTEMML-942?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Niketan Pansare reassigned SYSTEMML-942:
----------------------------------------
Assignee: Nakul Jindal
> Enable setting GPU from MLContext (and related APIs)
> ----------------------------------------------------
>
> Key: SYSTEMML-942
> URL: https://issues.apache.org/jira/browse/SYSTEMML-942
> Project: SystemML
> Issue Type: Sub-task
> Components: Compiler, Runtime
> Reporter: Niketan Pansare
> Assignee: Nakul Jindal
> Fix For: SystemML 0.11
>
>
> On setting GPU, MLContext needs to invoke:
> DMLScript.USE_ACCELERATOR = true;
> GPUContext.createGPUContext(); // Set GPU memory budget
> and upon exit, it should ec.destroyGPUContext();
> This logic should be enabled via Java/Python MLContext, MLPipeline/MLLearn
> wrappers.
> [~deron] [~mwdus...@us.ibm.com]
--
This message was sent by Atlassian JIRA
(v6.3.15#6346)
