refactor(serving): remove now-unused guard, debug, profiling, and host LM-head code from npu runners

[ndleslx]

Four focused dead-code cleanups, one per commit. Net deletion across all four; rebased onto current main.

refactor(deepseek): remove _validate_kernel_contract startup guard

Removes the fail-fast _validate_kernel_contract guard in DeepSeekV4PyptoExecutor plus its AST helper (_int_constant_from_file, _AST_INT_OPERATORS) and the now-unused import ast / import operator. The guard compared checked-in pypto-lib kernel constants against the serving layout at model load; purely defensive, no runtime dependency. Its two dedicated unit tests are dropped.

Also drops the now-unused importlib.util import, the unreferenced profile_verbose property (no callers; the base class does not define it), and the orphaned l3_trace constructor param / self._l3_trace attribute.

refactor(deepseek): remove npu_runner debug instrumentation

Removes the ad-hoc print-based debug scaffolding from DeepSeekV4ModelRunner:

• _debug_prefill_dispatch, _debug_decode_dispatch, _debug_tensor_stats, _debug_tensor_stats_enabled, _tensor_is_finite and all their call sites (prefill / decode / lm-head), plus the now-unused import os. Gated on undocumented PYPTO_DSV4_* env vars, referenced nowhere outside this file.
• The optional scope-stats profiling path (_scope_stats_run_config + its run_config branch in _run_l3), whose default output dir was hardcoded to a personal /data/liuxu/... path.
• The MTP decode test that monkeypatched _debug_decode_dispatch.

refactor(serving): drop unused L3 timing plumbing from npu runners

Both runners captured the worker.run() return value and attached host/device wall times to profiling spans via per-runner helpers (_kernel_trace_name, _add_run_timing_args in deepseek; _kernel_trace_name, _run_timing_us, _add_run_timing_args in qwen). The captured timing was no longer consumed, so the helpers are removed and the dispatch methods (DeepSeekV4ModelRunner._run_l3, Qwen314BModelRunner._run_distributed_program) are simplified to a plain worker.run(...) that no longer records or returns timing. Profiling span names now use callable_spec.name directly.

refactor(deepseek): remove dead host LM-head and worker-reset code

Removes production-dead methods from DeepSeekV4ModelRunner now that the LM head runs on device and these host paths have no callers:

• _is_layer_weight_name, _logits_for_hidden, _final_hidden, _final_norm (host-side LM-head / final-norm cluster)
• _invalidate_resident_cache_tensors, _reset_l3_worker (unused worker teardown helpers)
• the now-orphaned DEEPSEEK_V4_RMS_NORM_EPS / DEEPSEEK_V4_HC_EPS constants

Drops the two unit tests that exercised the removed host LM-head path.

Verification

• No dangling references to any removed symbol repo-wide.
• All touched modules py_compile cleanly.
• pytest tests/unit/model/ → 70 passed (on the rebased branch).

[coderabbitai]

Important

Review skipped

Auto incremental reviews are disabled on this repository.

Please check the settings in the CodeRabbit UI or the .coderabbit.yaml file in this repository. To trigger a single review, invoke the @coderabbitai review command.

⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: fbcb3b8d-77a3-44d0-b7ae-57acdfd6b71d

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

Use the checkbox below for a quick retry:

• 🔍 Trigger review

📝 Walkthrough

Walkthrough

DeepSeek kernel contract validation, profiling exposure, dispatch diagnostics, tensor checks, and optional scope-statistics execution were removed. Related unit-test cases and debug-hook setup were deleted.

Changes

DeepSeek runtime cleanup

Layer / File(s)	Summary
Remove kernel contract validation pypto_serving/model/deepseek/npu_executor.py, tests/unit/model/deepseek/test_model_components.py	The executor no longer parses kernel constants, validates required modules or configuration values, or exposes profile_verbose. Tests for contract rejection were removed.
Remove runner diagnostics pypto_serving/model/deepseek/npu_runner.py, tests/unit/model/deepseek/test_model_components.py	Dispatch hooks, tensor statistics, finite-output checks, LM-head diagnostics, and scope-statistics configuration were removed. L3 execution always calls worker.run with the compiled kernel and arguments. The fused MTP test no longer installs a debug hook.

Estimated code review effort: 3 (Moderate) | ~20 minutes

Possibly related PRs

• hw-native-sys/pypto-serving#135: Overlaps with removal of profile_verbose and optional L3 scope-statistics handling.
• hw-native-sys/pypto-serving#140: Modifies the same DeepSeek kernel-contract validation and related tests.
• hw-native-sys/pypto-serving#111: Overlaps with removal of DeepSeek profiling and debug instrumentation.

Poem

A rabbit hops through kernels bright,
Contract checks fade from sight.
Debug prints rest, tensors run clean,
L3 calls stay direct and lean.
“Less noise,” says Bunny, “what a scene!”

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Title check	✅ Passed	The title clearly summarizes the main cleanup changes in the DeepSeek NPU runners.
Description check	✅ Passed	The description directly explains the removed guards, debug code, profiling paths, host LM-head code, tests, and verification.

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands.}