Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Separating _RAFT_HOST and _RAFT_DEVICE macros #836

Merged
merged 3 commits into from
Sep 23, 2022
Merged

Separating _RAFT_HOST and _RAFT_DEVICE macros #836

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
a03add7
Separating _RAFT_HOST and _RAFT_DEVICE macros
cjnolet Sep 22, 2022
e35e088
Adding _FORCEINLINE
cjnolet Sep 22, 2022
458558c
Fixing style
cjnolet Sep 22, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions cpp/include/raft/core/detail/macros.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,12 +24,18 @@

#ifndef _RAFT_HOST_DEVICE
#if defined(_RAFT_HAS_CUDA)
#define _RAFT_HOST_DEVICE __host__ __device__
#define _RAFT_DEVICE __device__
#define _RAFT_HOST __host__
#define _RAFT_FORCEINLINE __forceinline__
#else
#define _RAFT_HOST_DEVICE
#define _RAFT_DEVICE
#define _RAFT_HOST
#define _RAFT_FORCEINLINE inline
#endif
#endif

#define _RAFT_HOST_DEVICE _RAFT_HOST _RAFT_DEVICE

#ifndef RAFT_INLINE_FUNCTION
#define RAFT_INLINE_FUNCTION inline _RAFT_HOST_DEVICE
#define RAFT_INLINE_FUNCTION _RAFT_FORCEINLINE _RAFT_HOST_DEVICE
#endif