Skip to content

feat: add gtest coverage for cuda act_and_mul activation kernel.#987

Merged
yingxudeng merged 1 commit intojd-opensource:mainfrom
yingxudeng:feat/gtest_active
Mar 9, 2026
Merged

feat: add gtest coverage for cuda act_and_mul activation kernel.#987
yingxudeng merged 1 commit intojd-opensource:mainfrom
yingxudeng:feat/gtest_active

Conversation

@yingxudeng
Copy link
Copy Markdown
Collaborator

No description provided.

Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

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

Code Review

This pull request adds gtest coverage for the act_and_mul CUDA kernel. The test implementation is solid, covering different activation modes and input dimensions to target various kernel code paths. I've identified a few areas for improvement: a missing linker dependency in the CMakeLists.txt file which could cause build failures, and an opportunity to extend test coverage to include the bfloat16 data type, which is supported by the kernel. I've left specific suggestions to address these points.

Comment thread xllm/core/kernels/cuda/CMakeLists.txt
Comment thread xllm/core/kernels/cuda/activation_test.cpp
Comment thread xllm/core/kernels/cuda/activation_test.cpp Outdated
Copy link
Copy Markdown
Collaborator

@yq33victor yq33victor left a comment

Choose a reason for hiding this comment

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

LGTM

@yingxudeng yingxudeng merged commit f9e0780 into jd-opensource:main Mar 9, 2026
28 of 32 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants