no need to convert

a2f2986a · rusty1s · 56ae40e7 · a2f2986a
Commit a2f2986a authored Jan 13, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torch_cluster/src/generic/cpu.c torch_cluster/src/generic/cpu.c +1 -1

No files found.
--- a/torch_cluster/src/generic/cpu.c
+++ b/torch_cluster/src/generic/cpu.c
@@ -11,7 +11,7 @@ void cluster_(grid)(int C, THLongTensor *output, THTensor *position, THTensor *s
    tmp = C; c = 0;
    for (i = 0; i < d; i++) {
      tmp = tmp / *(count_data + i);
-      c += tmp * (int64_t)floor((float)(*(position_data + i * position_stride) / *(size_data + i)));
+      c += tmp * (int64_t)floor(*(position_data + i * position_stride) / *(size_data + i));
    }
    output_data[0] = c;
  )