Unverified Commit 451709db authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #37 from rusty1s/pytorch20

PyTorch 2.0 fix
parents 731f8cd2 55abd231
...@@ -30,4 +30,4 @@ int64_t cuda_version() { ...@@ -30,4 +30,4 @@ int64_t cuda_version() {
} }
static auto registry = torch::RegisterOperators().op( static auto registry = torch::RegisterOperators().op(
"torch_spline_conv::cuda_version", &cuda_version); "torch_spline_conv::cuda_version", [] { return cuda_version(); });
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment