Skip to content

Add LP-type-aware Torch-JIT surrogate models #241

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
sanjaychari wants to merge 27 commits into
base: master
Choose a base branch
from
Draft

Add LP-type-aware Torch-JIT surrogate models #241

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
976bb79
MPI Replay: remove print_surrogate_stats() to compile cleanly
kevinabrown Jun 17, 2024
7f42f4a
director-b: started adding director LP for mpi-replay
kevinabrown Jul 5, 2024
5ae2e7c
director-b: complete initial director LP prototype for mpi-replay
kevinabrown Aug 16, 2024
138f46a
zmqml: update zmq server and requester to interface with director LP
kevinabrown Aug 16, 2024
39fbc4f
Fix zmq and ROSS compilation issues
sanjaychari May 21, 2026
0651b5e
Fix torch-jit compilation
sanjaychari May 21, 2026
01a2b16
Allow cpu-based PyTorch usage
sanjaychari May 21, 2026
51f691b
Add ML models
sanjaychari May 22, 2026
e42e75a
Move ML models to surrogate directory
sanjaychari May 27, 2026
4a50395
surrogate: support LP-aware Torch-JIT latency model
sanjaychari May 27, 2026
5534e2c
surrogate: add Torch-JIT debug diagnostics
sanjaychari May 27, 2026
3c3f78d
Improve debug print check mechanism
sanjaychari May 27, 2026
399ec0f
Add LP-type-aware Torch-JIT surrogate models
sanjaychari May 28, 2026
47d9d84
Make zeromq dependency handling robust
sanjaychari Jun 1, 2026
81deb94
Fix installation with swm and union
sanjaychari Jun 2, 2026
36f8486
Add Zeromq director example configs
sanjaychari Jun 2, 2026
d970c3e
Add requirements.txt file for surrogate
sanjaychari Jun 3, 2026
aea6585
Add iteration time ML model for zmq-based director
sanjaychari Jun 4, 2026
06f1fb9
Add optimistic RC for ZeroMQ Director
sanjaychari Jun 4, 2026
9bc7aa1
Add flag to shutdown zeromq server after a run
sanjaychari Jun 5, 2026
5d6fb19
Introduce pretraining for zmq hybrid simulations
sanjaychari Jun 8, 2026
76545e9
Zmq Director: Fix unmatched sends/recvs issue
sanjaychari Jun 9, 2026
1e04405
Add template conf file for milc
sanjaychari Jun 9, 2026
a2cd82b
Add Director retraining workflow
sanjaychari Jun 9, 2026
e08ecb6
Fix union compilation issue with python-override
sanjaychari Jun 10, 2026
4163cdf
Fix CUDA arch detection issue
sanjaychari Jun 11, 2026
0584d9f
Improve iteration ml model and add union confs
sanjaychari Jun 11, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,8 @@
/maint/codes.pc
/test-driver
.deps
src/surrogate/zmqml/demozmqmlrequester
src/surrogate/zmqml/libzmqmlrequester.so

# make generated artifacts
.dirstamp
Expand All @@ -44,3 +46,4 @@ install-mastiff/include/codes/model-net-method.h
/build*
.cache
compile_commands.json
__pycache__/
Loading