-
Notifications
You must be signed in to change notification settings - Fork 183
Add UCX Plugin Gtests #338
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.
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
base: main
Are you sure you want to change the base?
Conversation
|
👋 Hi glevnv! Thank you for contributing to ai-dynamo/nixl. Your PR reviewers will review your contribution then trigger the CI to test your changes. 🚀 |
e905956 to
371562f
Compare
aa50c48 to
2673c3f
Compare
c33714a to
248e595
Compare
Add a plugin gtest infrastructure to test baackend plugin local xfer, remote xfer, progress and notification handling. Add ucx plugin gtests on top of the plugin infrastructure.
| } | ||
| }; | ||
|
|
||
| #ifdef HAVE_CUDA |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please align with @dfarge implementation, allowing to omit these configuration flags from the code
| }; | ||
|
|
||
| template<> struct MemoryHandler<DRAM_SEG> { | ||
| static void * |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Seems that these structs are stateless, maybe a namespace can replace?
|
|
||
| static void | ||
| releasePtr(void *addr) { | ||
| std::cerr << "Unsupported memory type!" << std::endl; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We have a dedicated logger
| static void | ||
| releasePtr(void *addr) { | ||
| std::cerr << "Unsupported memory type!" << std::endl; | ||
| assert(0); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can abseil::CHECK replace? (or one of its flavors)
| static const size_t NUM_BUF_ENTRIES = 64; | ||
| static const size_t BUF_ENTRY_SIZE = 1024 * 1024; | ||
| static const size_t BUF_SIZE = BUF_ENTRY_SIZE * NUM_BUF_ENTRIES; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
constexpr
|
Any update on this @glevnv? |
Adds a plugin gtest infrastructure.
Adds ucx plugin gtests on top of the plugin infrastructure.