Skip to content

Commit e7a9905

Browse files
jiangyangmuThe tunix Authors
authored andcommitted
Rewrite perf tracer with new data model
PiperOrigin-RevId: 834051083
1 parent 40c2577 commit e7a9905

File tree

9 files changed

+1104
-746
lines changed

9 files changed

+1104
-746
lines changed

tests/perf/trace_test.py

Lines changed: 229 additions & 171 deletions
Large diffs are not rendered by default.

tunix/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,8 @@
3131
from tunix.generate.sampler import Sampler
3232
from tunix.generate.tokenizer_adapter import TokenizerAdapter, Tokenizer
3333
from tunix.perf.export import PerfMetricsExport
34-
from tunix.perf.metrics import PerfMetricsApi
3534
from tunix.perf.metrics import PerfMetricsConfig
36-
from tunix.perf.metrics import PerfMetricsQuery
35+
from tunix.perf.metrics import PerfSpanQuery
3736
from tunix.rl.grpo.grpo_learner import GRPOConfig
3837
from tunix.rl.grpo.grpo_learner import GrpoConfig
3938
from tunix.rl.grpo.grpo_learner import GRPOLearner

0 commit comments

Comments
 (0)