Skip to content

Conversation

@glevnv
Copy link
Contributor

@glevnv glevnv commented May 20, 2025

Adds a plugin gtest infrastructure.
Adds ucx plugin gtests on top of the plugin infrastructure.

@copy-pr-bot
Copy link

copy-pr-bot bot commented May 20, 2025

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@github-actions
Copy link

👋 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.

🚀

@glevnv glevnv force-pushed the ucx-gtests branch 13 times, most recently from e905956 to 371562f Compare May 20, 2025 14:27
@glevnv glevnv force-pushed the ucx-gtests branch 10 times, most recently from aa50c48 to 2673c3f Compare May 21, 2025 14:40
@glevnv glevnv force-pushed the ucx-gtests branch 9 times, most recently from c33714a to 248e595 Compare May 22, 2025 11:11
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.
@glevnv glevnv marked this pull request as ready for review May 22, 2025 13:01
}
};

#ifdef HAVE_CUDA
Copy link
Contributor

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 *
Copy link
Contributor

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;
Copy link
Contributor

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);
Copy link
Contributor

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)

Comment on lines +36 to +38
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;
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

constexpr

@aranadive
Copy link
Contributor

Any update on this @glevnv?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants