aparler/PyNorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

first commit distributed module

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-23 19:15:30 -03:00
parent 55fc1ac020
commit b152ae2a1d
10 changed files with 83 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
norch/csrc/tensor.cpp
View file

@ -64,8 +64,15 @@ extern "C" {
}
void to_device(Tensor* tensor, char* target_device) {
int device_id = 0;
char* endptr;
long num = strtol(target_device, &endptr, 10);
if (*endptr == '\0') {
device_id = (int)num;
}
if ((strcmp(target_device, "cuda") == 0) && (strcmp(tensor->device, "cpu") == 0)) {
cpu_to_cuda(tensor);
cpu_to_cuda(tensor, device_id);
}
else if ((strcmp(target_device, "cpu") == 0) && (strcmp(tensor->device, "cuda") == 0)) {