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

small fix max cuda axis was setting init to 0 instead of -inf

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-23 01:29:23 -03:00
parent 5882f1d801
commit e2a0262c4e
1 changed files with 0 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
norch/csrc/cuda.cu
View file

@ -328,7 +328,6 @@ __host__ void max_tensor_cuda(Tensor* tensor, float* result_data, int axis) {
// Initialize result_data to 0
float neg_inf = -FLT_MAX;
cudaMemset(result_data, *reinterpret_cast<int*>(&neg_inf), result_size * sizeof(float));
cudaMemset(result_data, 0, result_size * sizeof(float));
int num_threads = result_size;
int num_blocks = (num_threads + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK;