Skip to content

Feature/add gpu lightgbm estimator#50

Merged
yoshifuminakamura merged 4 commits into
developfrom
feature/add-gpu-lightgbm-estimator
Jun 16, 2026
Merged

Feature/add gpu lightgbm estimator#50
yoshifuminakamura merged 4 commits into
developfrom
feature/add-gpu-lightgbm-estimator

Conversation

@yoshifuminakamura

Copy link
Copy Markdown
Collaborator

No description provided.

Connect PerfTools LightGBM_model/1.0 as a GPU kernel section package so GENESIS can exercise the second GPU estimation model through the existing instrumented section flow.

Add a small NCU raw CSV preparation bridge for LightGBM, wire the shared PerfTools checkout into the estimation runner, and make GENESIS default to the LightGBM GPU section package while still allowing BK_GENESIS_GPU_SECTION_PACKAGE to select the MLP path.

Keep padata archives smaller by excluding heavy Nsight Compute binary reports by default. The raw CSV and text import remain available for estimation and portal summaries, with BK_PROFILER_ARCHIVE_NCU_REPORT=true left as an explicit debugging opt-in.

Add shell coverage for the LightGBM prediction CSV path, GENESIS package selection, and the new profiler archive behavior.

Signed-off-by: Yoshifumi Nakamura <nakamura@riken.jp>
Propagate section package transform failures through the top-level breakdown dispatcher so external predictor failures cannot produce a green estimate job with missing values.

Make the MLP and LightGBM predictor wrappers fail explicitly when inference exits non-zero or no prediction CSV is produced, and cover the LightGBM failure path in shell tests.

Signed-off-by: Yoshifumi Nakamura <nakamura@riken.jp>
Treat recorded/current weak-scaling breakdowns as identity/logp projections instead of carrying cross-system GPU section packages into the current side.

This removes misleading fallback metadata such as gpu_kernel_lightgbm_v10 -> identity from current_system while preserving the future-side GPU estimator result.

Signed-off-by: Yoshifumi Nakamura <nakamura@riken.jp>
Document GPU kernel estimator packages as selectable options rather than canonical defaults. Note that GENESIS can choose the section package through BK_GENESIS_GPU_SECTION_PACKAGE during bring-up, and keep qws smoke guidance scoped to the MLP package.

Signed-off-by: Yoshifumi Nakamura <nakamura@riken.jp>
@yoshifuminakamura yoshifuminakamura merged commit 9a467c5 into develop Jun 16, 2026
5 checks passed
@yoshifuminakamura yoshifuminakamura deleted the feature/add-gpu-lightgbm-estimator branch June 16, 2026 23:53
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.

1 participant