Skip to content

GitLab

You need to sign in or sign up before continuing.
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
Show 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