File tree 1 file changed +4
-6
lines changed
1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change 1
1
#ifdef WITH_PYTHON
2
2
#include < Python.h>
3
3
#endif
4
- #include < torch/script.h>
5
4
#include " cluster.h"
6
5
#include " macros.h"
6
+ #include < torch/script.h>
7
7
8
8
#ifdef WITH_CUDA
9
9
#ifdef USE_ROCM
@@ -23,7 +23,6 @@ PyMODINIT_FUNC PyInit__version_cpu(void) { return NULL; }
23
23
#endif
24
24
#endif
25
25
26
-
27
26
namespace cluster {
28
27
CLUSTER_API int64_t cuda_version () noexcept {
29
28
#ifdef WITH_CUDA
@@ -36,8 +35,7 @@ CLUSTER_API int64_t cuda_version() noexcept {
36
35
return -1 ;
37
36
#endif
38
37
}
39
- } // namespace sparse
40
-
38
+ } // namespace cluster
41
39
42
- static auto registry =
43
- torch::RegisterOperators ().op( " torch_cluster::cuda_version" , & cluster::cuda_version);
40
+ static auto registry = torch::RegisterOperators().op(
41
+ " torch_cluster::cuda_version" , [] { return cluster::cuda_version (); } );
You can’t perform that action at this time.
0 commit comments