[
https://issues.apache.org/jira/browse/SYSTEMML-942?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15898703#comment-15898703
]
Nakul Jindal commented on SYSTEMML-942:
---------------------------------------
Opened [PR 942|https://github.com/apache/incubator-systemml/pull/420]
> 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 1.0
>
>
> 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)
