Skip to content

Commit 6b87660

Browse files
bottlerfacebook-github-bot
authored andcommitted
Use cuda's make_float3 in pulsar
Summary: Fixes github.com/facebookresearch/issues/1680 Reviewed By: MichaelRamamonjisoa Differential Revision: D51587889 fbshipit-source-id: e68ae32d7041fb9ea3e981cf2bde47f947a41ca2
1 parent c373a84 commit 6b87660

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

pytorch3d/csrc/pulsar/global.h

+1-7
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@
5656
#pragma clang diagnostic pop
5757
#ifdef WITH_CUDA
5858
#include <ATen/cuda/CUDAContext.h>
59+
#include <vector_functions.h>
5960
#else
6061
#ifndef cudaStream_t
6162
typedef void* cudaStream_t;
@@ -74,13 +75,6 @@ struct float3 {
7475
};
7576
#endif
7677
namespace py = pybind11;
77-
inline float3 make_float3(const float& x, const float& y, const float& z) {
78-
float3 res;
79-
res.x = x;
80-
res.y = y;
81-
res.z = z;
82-
return res;
83-
}
8478

8579
inline bool operator==(const float3& a, const float3& b) {
8680
return a.x == b.x && a.y == b.y && a.z == b.z;

0 commit comments

Comments
 (0)