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

Fix sum cuda reduce

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-02 13:22:52 -03:00
parent 6b1ea4be88
commit 7fb2ee9072
6 changed files with 12 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

3
norch/csrc/tensor.cpp
View file

@ -10,7 +10,7 @@
extern "C" {
Tensor* create_tensor(float* data, int* shape, int ndim, char* device) {
printf("Creating tensor\n");
Tensor* tensor = (Tensor*)malloc(sizeof(Tensor));
if (tensor == NULL) {
@ -172,7 +172,6 @@ extern "C" {
float* result_data;
cudaMalloc((void**)&result_data, tensor->size * sizeof(float));
result_data[0] = 0;
sum_tensor_cuda(tensor, result_data);
return create_tensor(result_data, shape, ndim, device);
}