Skip to content

Move cost calculation to backend#544

Merged
corbt merged 1 commit intomainfrom
cost-calculator-backend
Jan 31, 2026
Merged

Move cost calculation to backend#544
corbt merged 1 commit intomainfrom
cost-calculator-backend

Conversation

@corbt
Copy link
Copy Markdown
Collaborator

@corbt corbt commented Jan 31, 2026

Summary

  • Introduce centralized cost utilities and a model-level cost calculator (default no-op)
  • Compute training costs in TinkerNative backend and remove trainer-side cost inference
  • Aggregate arbitrary costs_* components; reject manual costs_total
  • Log costs at current step with an instance-scoped lock
  • Update binary prefix pipeline to use the model cost calculator

Testing

  • pre-commit hooks via commit (ruff, ruff-format, ty, uv.lock sync)

@corbt corbt merged commit 97938fb into main Jan 31, 2026
2 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.

1 participant