Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
rusty1s committed Jan 11, 2025
1 parent 975d8ca commit 2df8118
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions pyg_lib/csrc/classes/cpu/hash_map.h
Original file line number Diff line number Diff line change
Expand Up @@ -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",
[&] {
Expand All @@ -36,8 +35,7 @@ struct CPUHashMap : torch::CustomClassHolder {
auto out_data = out.data_ptr<int64_t>();

// 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",
[&] {
Expand Down

0 comments on commit 2df8118

Please sign in to comment.