[jira] [Commented] (SINGA-41) Support single node single GPU training

[ASF subversion and git services (JIRA)]

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

ASF subversion and git services commented on SINGA-41:
------------------------------------------------------

Commit 2818605ac10a28af542d764156588d65737f18fd in incubator-singa's branch 
refs/heads/gpu from seaok
[ https://git-wip-us.apache.org/repos/asf?p=incubator-singa.git;h=2818605 ]

SINGA-41:Support single node single GPU training

Rebase to lastest SINGA master


> Support single node single GPU  training
> ----------------------------------------
>
>                 Key: SINGA-41
>                 URL: https://issues.apache.org/jira/browse/SINGA-41
>             Project: Singa
>          Issue Type: New Feature
>            Reporter: Haibo Chen
>
> For new feature: support single node single GPU training.
> The blob class:
> 1) add a member function
> Dtype* mutable_xpu_data();  // make a decision(GPU or CPU) at compile time
> 2) add a include file: #include <cuda_runtime.h> //support cuda code
> The param class:
> 1) add three member function
> float* mutable_xpu_data() 
> float* mutable_xpu_grad() 
> float* mutable_xpu_history() 
> The layer class:
> 1) we use xpu instead of cpu in some layers of 
> subclass,eg.ConvolutionLayer、Dropout...
> Makefile.GPU:
> 1) For CPU mode,set CUDA_DIR :=
> step 1:
> make
> 2) For GPU mode,set CUDA_DIR := your cuda installation path
> step 1:
> make gpu         // rename layer file : layer.cc ->  layer.cu
> step 2:
> make



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