You are viewing a plain text version of this content. The canonical link for it is here.

Posted to issues@systemml.apache.org by "Niketan Pansare (JIRA)" <ji...@apache.org> on 2016/09/19 04:28:20 UTC

[jira] [Created] (SYSTEMML-942) Enable setting GPU from MLContext (and related APIs)

Niketan Pansare created SYSTEMML-942:
----------------------------------------

             Summary: 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
            Reporter: Niketan Pansare


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] [~mwdusenb@us.ibm.com]



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