Commit aa7c667
Fix compilation errors and test failures across tpu_sync targets
- Qualify PollSaveStatusResult and include kv_cache_store.h in
kv_cache_store_mock.h/cc, adding missing kv_cache dependency.
- Rename mock registration macro in kv_cache_store_backend_mock.h to
REGISTER_MOCK_KV_CACHE_STORE_BACKEND to prevent macro collision with
the production kv_cache_store_backend_factory.h macro.
- Add torch_tpu_device_buffer_headers dependency to
weight_synchronizer_torch_mock to fix layering check failure.
- Update PollSaveStatus() structured bindings in storage_offload_test.cc
to unpack all 5 fields of PollSaveStatusResult.
- Add default metric instance registration in RegisterKnownFamilies() in
prometheus_exporter.cc to ensure text snapshot serializes registered
metric families on initialization.
- Configure pathways_qwen_weight_synchronization_perf_test with reduced
layers/iterations and appropriate perf/hardware test tags (notap,
cpu:8, requires-accel-hbm, requires-net:external).
Tested:
- blaze test --nocheck_visibility //third_party/tpu_raiden/tpu_sync/experimental/feat_storage_extension:kv_cache_store_mock
- blaze test --nocheck_visibility //third_party/tpu_raiden/tpu_sync/experimental/feat_storage_extension:storage_offload_test
- blaze test --nocheck_visibility //third_party/tpu_raiden/tpu_sync/frameworks/torch:weight_synchronizer_torch_mock
- blaze test --nocheck_visibility //third_party/tpu_raiden/tpu_sync/telemetry/python:telemetry_binding_test
- blaze test --nocheck_visibility //third_party/tpu_raiden/tpu_sync/telemetry:prometheus_exporter_test
- Full test pass across //third_party/tpu_raiden/tpu_sync/...
PiperOrigin-RevId: 9699548751 parent d07d955 commit aa7c667
2 files changed
Lines changed: 4 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
429 | 429 | | |
430 | 430 | | |
431 | 431 | | |
| 432 | + | |
432 | 433 | | |
433 | 434 | | |
434 | 435 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
75 | 75 | | |
76 | 76 | | |
77 | 77 | | |
| 78 | + | |
78 | 79 | | |
79 | 80 | | |
80 | 81 | | |
| |||
83 | 84 | | |
84 | 85 | | |
85 | 86 | | |
| 87 | + | |
86 | 88 | | |
87 | 89 | | |
88 | 90 | | |
| |||
91 | 93 | | |
92 | 94 | | |
93 | 95 | | |
| 96 | + | |
94 | 97 | | |
95 | 98 | | |
96 | 99 | | |
| |||
0 commit comments