From 2df811857694127adc9ae88f32a227c3f952a1c7 Mon Sep 17 00:00:00 2001 From: rusty1s Date: Sat, 11 Jan 2025 08:29:23 +0000 Subject: [PATCH] update --- pyg_lib/csrc/classes/cpu/hash_map.h | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/pyg_lib/csrc/classes/cpu/hash_map.h b/pyg_lib/csrc/classes/cpu/hash_map.h index ba32f1b3..4896dd35 100644 --- a/pyg_lib/csrc/classes/cpu/hash_map.h +++ b/pyg_lib/csrc/classes/cpu/hash_map.h @@ -14,8 +14,7 @@ struct CPUHashMap : torch::CustomClassHolder { // TODO Assert 1-dim // clang-format off - AT_DISPATCH_ALL_TYPES_AND( - at::ScalarType::Bool, + AT_DISPATCH_ALL_TYPES_AND(at::ScalarType::Bool, key.scalar_type(), "cpu_hash_map_init", [&] { @@ -36,8 +35,7 @@ struct CPUHashMap : torch::CustomClassHolder { auto out_data = out.data_ptr(); // clang-format off - AT_DISPATCH_ALL_TYPES_AND( - at::ScalarType::Bool, + AT_DISPATCH_ALL_TYPES_AND(at::ScalarType::Bool, query.scalar_type(), "cpu_hash_map_get", [&] {