Skip to content

GitLab

Unverified Commit 4d8ae71b authored by Muhammed Fatih BALIN's avatar Muhammed Fatih BALIN Committed by GitHub
Browse files

[GraphBolt][CUDA] Macro for all types (#6824)

parent 26e740a3
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 0 deletions
graphbolt/src/cuda/common.h
View file @ 4d8ae71b
...@@ -90,6 +90,16 @@ inline bool is_zero<dim3>(dim3 size) { ...@@ -90,6 +90,16 @@ inline bool is_zero<dim3>(dim3 size) {
} \ } \
} }
// This includes all integer, float and boolean types.
#define GRAPHBOLT_DISPATCH_CASE_ALL_TYPES(...) \
AT_DISPATCH_CASE_ALL_TYPES(__VA_ARGS__) \
AT_DISPATCH_CASE(at::ScalarType::Half, __VA_ARGS__) \
AT_DISPATCH_CASE(at::ScalarType::BFloat16, __VA_ARGS__) \
AT_DISPATCH_CASE(at::ScalarType::Bool, __VA_ARGS__)
#define GRAPHBOLT_DISPATCH_ALL_TYPES(TYPE, NAME, ...) \
AT_DISPATCH_SWITCH(TYPE, NAME, GRAPHBOLT_DISPATCH_CASE_ALL_TYPES(__VA_ARGS__))
#define GRAPHBOLT_DISPATCH_ELEMENT_SIZES(element_size, name, ...) \ #define GRAPHBOLT_DISPATCH_ELEMENT_SIZES(element_size, name, ...) \
[&] { \ [&] { \
switch (element_size) { \ switch (element_size) { \
......
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