Fix CPU hashing (#173)

kentang-mit · web-flow · commit 6719899f1acf · 2022-09-26T18:19:15.000-04:00
diff --git a/torchsparse/backend/hash/hash_cpu.cpp b/torchsparse/backend/hash/hash_cpu.cpp
@@ -26,7 +26,7 @@ void cpu_kernel_hash_wrapper(int N, int K, const int *data,
       for (int j = 0; j < 3; j++) {
         cur_coord[j] = data[i * 4 + j] + kernel_offset[k * 3 + j];
       }
-      cur_coord[3] = data[3];
+      cur_coord[3] = data[i * 4 + 3];
       uint64_t hash = 14695981039346656037UL;
       for (int j = 0; j < 4; j++) {
         hash ^= (unsigned int)cur_coord[j];

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ void cpu_kernel_hash_wrapper(int N, int K, const int *data,`
`26`	`26`	`for (int j = 0; j < 3; j++) {`
`27`	`27`	`cur_coord[j] = data[i * 4 + j] + kernel_offset[k * 3 + j];`
`28`	`28`	`}`
`29`		`- cur_coord[3] = data[3];`
	`29`	`+ cur_coord[3] = data[i * 4 + 3];`
`30`	`30`	`uint64_t hash = 14695981039346656037UL;`
`31`	`31`	`for (int j = 0; j < 4; j++) {`
`32`	`32`	`hash ^= (unsigned int)cur_coord[j];`