chialoops 1.0.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (214) hide show
  1. chialoops-1.0.0/LICENSE +29 -0
  2. chialoops-1.0.0/PKG-INFO +149 -0
  3. chialoops-1.0.0/README.md +100 -0
  4. chialoops-1.0.0/chia/analysis/__init__.py +1 -0
  5. chialoops-1.0.0/chia/analysis/sustainability/__init__.py +23 -0
  6. chialoops-1.0.0/chia/analysis/sustainability/focal.py +472 -0
  7. chialoops-1.0.0/chia/analysis/sustainability/pfas.py +127 -0
  8. chialoops-1.0.0/chia/aws/__init__.py +0 -0
  9. chialoops-1.0.0/chia/aws/config.py +98 -0
  10. chialoops-1.0.0/chia/aws/ec2.py +333 -0
  11. chialoops-1.0.0/chia/aws/host.py +76 -0
  12. chialoops-1.0.0/chia/aws/s3.py +350 -0
  13. chialoops-1.0.0/chia/aws/test/test_s3.py +366 -0
  14. chialoops-1.0.0/chia/aws/test/test_s3_e2e.py +153 -0
  15. chialoops-1.0.0/chia/base/ChiaFunction.py +711 -0
  16. chialoops-1.0.0/chia/base/bypass.py +442 -0
  17. chialoops-1.0.0/chia/base/cache.py +430 -0
  18. chialoops-1.0.0/chia/base/chia_kv_store.py +199 -0
  19. chialoops-1.0.0/chia/base/chia_wait.py +331 -0
  20. chialoops-1.0.0/chia/base/colocated.py +204 -0
  21. chialoops-1.0.0/chia/base/dispatch_proxy.py +247 -0
  22. chialoops-1.0.0/chia/base/llm_call.py +93 -0
  23. chialoops-1.0.0/chia/base/pid_registry.py +471 -0
  24. chialoops-1.0.0/chia/base/test/ChiaFunction_tb.py +272 -0
  25. chialoops-1.0.0/chia/base/test/chia_wait_tb.py +186 -0
  26. chialoops-1.0.0/chia/base/test/pid_registry_tb.py +743 -0
  27. chialoops-1.0.0/chia/base/test/ray_job_stop_helper.py +76 -0
  28. chialoops-1.0.0/chia/base/test/sigterm_cleanup_helper.py +74 -0
  29. chialoops-1.0.0/chia/base/test/test_bypass_tags.py +479 -0
  30. chialoops-1.0.0/chia/base/test/test_cache_tags.py +520 -0
  31. chialoops-1.0.0/chia/base/test/test_chia_remote_blocking.py +140 -0
  32. chialoops-1.0.0/chia/base/test/test_colocated_live.py +264 -0
  33. chialoops-1.0.0/chia/base/test/test_dispatch_proxy_live.py +76 -0
  34. chialoops-1.0.0/chia/base/tools/AsyncBashTool.py +98 -0
  35. chialoops-1.0.0/chia/base/tools/AsyncJobTool.py +119 -0
  36. chialoops-1.0.0/chia/base/tools/BashTool.py +79 -0
  37. chialoops-1.0.0/chia/base/tools/ChiaTool.py +378 -0
  38. chialoops-1.0.0/chia/base/tools/ChiaToolTemplate.py +62 -0
  39. chialoops-1.0.0/chia/base/tools/test/test_tool.py +395 -0
  40. chialoops-1.0.0/chia/base/tools/util.py +25 -0
  41. chialoops-1.0.0/chia/chipyard/__init__.py +0 -0
  42. chialoops-1.0.0/chia/chipyard/chipyard_hammer.py +253 -0
  43. chialoops-1.0.0/chia/chipyard/chisel_build_node.py +540 -0
  44. chialoops-1.0.0/chia/chipyard/circt.py +834 -0
  45. chialoops-1.0.0/chia/chipyard/cosim_node.py +100 -0
  46. chialoops-1.0.0/chia/chipyard/firemarshal_node.py +358 -0
  47. chialoops-1.0.0/chia/chipyard/macrocompiler.py +371 -0
  48. chialoops-1.0.0/chia/chipyard/numa_prefix.py +86 -0
  49. chialoops-1.0.0/chia/chipyard/riscv_build_node.py +204 -0
  50. chialoops-1.0.0/chia/chipyard/riscv_dv_gen_node.py +381 -0
  51. chialoops-1.0.0/chia/chipyard/riscv_objdump_node.py +164 -0
  52. chialoops-1.0.0/chia/chipyard/spike_build_node.py +299 -0
  53. chialoops-1.0.0/chia/chipyard/state_def.py +456 -0
  54. chialoops-1.0.0/chia/chipyard/test/cluster/macrocompiler_e2e_test.py +351 -0
  55. chialoops-1.0.0/chia/chipyard/test/riscv_dv_gen_e2e_driver.py +105 -0
  56. chialoops-1.0.0/chia/chipyard/test/spike_cosim_e2e_driver.py +71 -0
  57. chialoops-1.0.0/chia/chipyard/test/test_build_program.py +43 -0
  58. chialoops-1.0.0/chia/chipyard/test/test_chipyard_hammer_live.py +319 -0
  59. chialoops-1.0.0/chia/chipyard/test/test_chisel_riscv_verilator_e2e.py +223 -0
  60. chialoops-1.0.0/chia/chipyard/test/test_firemarshal_node.py +105 -0
  61. chialoops-1.0.0/chia/chipyard/test/test_firesim_metasim_run.py +112 -0
  62. chialoops-1.0.0/chia/chipyard/test/test_riscv_build_failure.py +88 -0
  63. chialoops-1.0.0/chia/chipyard/test/test_riscv_timeout_output_decode.py +45 -0
  64. chialoops-1.0.0/chia/chipyard/test/test_spike_build_node.py +164 -0
  65. chialoops-1.0.0/chia/chipyard/test/test_spike_golden_model.py +110 -0
  66. chialoops-1.0.0/chia/chipyard/test/torture_e2e_driver.py +180 -0
  67. chialoops-1.0.0/chia/chipyard/test/torture_run_node_tb.py +182 -0
  68. chialoops-1.0.0/chia/chipyard/torture_run_node.py +407 -0
  69. chialoops-1.0.0/chia/chipyard/verilator_run_node.py +721 -0
  70. chialoops-1.0.0/chia/cli/__init__.py +0 -0
  71. chialoops-1.0.0/chia/cli/down.py +143 -0
  72. chialoops-1.0.0/chia/cli/firesim_cmds.py +297 -0
  73. chialoops-1.0.0/chia/cli/job.py +81 -0
  74. chialoops-1.0.0/chia/cli/main.py +276 -0
  75. chialoops-1.0.0/chia/cli/ray_passthrough.py +58 -0
  76. chialoops-1.0.0/chia/cli/up.py +481 -0
  77. chialoops-1.0.0/chia/cli/viz_cmd.py +29 -0
  78. chialoops-1.0.0/chia/cli/viz_profile_cmd.py +55 -0
  79. chialoops-1.0.0/chia/cluster/__init__.py +0 -0
  80. chialoops-1.0.0/chia/cluster/aws_nodes.py +784 -0
  81. chialoops-1.0.0/chia/cluster/config.py +1013 -0
  82. chialoops-1.0.0/chia/cluster/docker.py +136 -0
  83. chialoops-1.0.0/chia/cluster/gcp_nodes.py +762 -0
  84. chialoops-1.0.0/chia/cluster/log.py +45 -0
  85. chialoops-1.0.0/chia/cluster/node_setup.py +1200 -0
  86. chialoops-1.0.0/chia/cluster/ray_stop.py +450 -0
  87. chialoops-1.0.0/chia/cluster/ssh.py +239 -0
  88. chialoops-1.0.0/chia/cluster/tailnet.py +548 -0
  89. chialoops-1.0.0/chia/cluster/test/down/test_scoped_down_live.py +206 -0
  90. chialoops-1.0.0/chia/cluster/test/multi_cluster/test_multi_cluster_add.py +136 -0
  91. chialoops-1.0.0/chia/cluster/test/tailnet/__init__.py +0 -0
  92. chialoops-1.0.0/chia/cluster/test/tailnet/test_tailnet.py +692 -0
  93. chialoops-1.0.0/chia/cluster/test/tunnel/test_aws_nodes.py +227 -0
  94. chialoops-1.0.0/chia/cluster/test/tunnel/test_cross_machine.py +393 -0
  95. chialoops-1.0.0/chia/cluster/test/tunnel/test_dispatch_proxy.py +177 -0
  96. chialoops-1.0.0/chia/cluster/test/tunnel/test_ec2_cluster.py +55 -0
  97. chialoops-1.0.0/chia/cluster/test/tunnel/test_ec2_concurrency.py +231 -0
  98. chialoops-1.0.0/chia/cluster/test/tunnel/test_ec2_tool.py +400 -0
  99. chialoops-1.0.0/chia/cluster/test/tunnel/test_gcp_cluster.py +60 -0
  100. chialoops-1.0.0/chia/cluster/test/tunnel/test_gcp_nodes.py +492 -0
  101. chialoops-1.0.0/chia/cluster/tunnel.py +281 -0
  102. chialoops-1.0.0/chia/cluster/worker_provisioner.py +1141 -0
  103. chialoops-1.0.0/chia/database/__init__.py +25 -0
  104. chialoops-1.0.0/chia/database/base.py +334 -0
  105. chialoops-1.0.0/chia/database/postgres_node.py +715 -0
  106. chialoops-1.0.0/chia/database/sqlite_node.py +672 -0
  107. chialoops-1.0.0/chia/database/test/test_postgres_node_live.py +527 -0
  108. chialoops-1.0.0/chia/database/test/test_sqlite_node_live.py +510 -0
  109. chialoops-1.0.0/chia/esp/__init__.py +0 -0
  110. chialoops-1.0.0/chia/esp/esp_workspace.py +1111 -0
  111. chialoops-1.0.0/chia/esp/state_def.py +203 -0
  112. chialoops-1.0.0/chia/esp/test/cluster/accel_e2e_test.py +156 -0
  113. chialoops-1.0.0/chia/esp/test/cluster/accel_loop_e2e_test.py +150 -0
  114. chialoops-1.0.0/chia/esp/test/cluster/fpga_e2e_test.py +168 -0
  115. chialoops-1.0.0/chia/esp/test/cluster/hls_loop_e2e_test.py +161 -0
  116. chialoops-1.0.0/chia/esp/test/cluster/soft_build_e2e_test.py +149 -0
  117. chialoops-1.0.0/chia/esp/test/cluster/xmsim_e2e_test.py +112 -0
  118. chialoops-1.0.0/chia/esp/test/test_esp_local.py +607 -0
  119. chialoops-1.0.0/chia/firesim/__init__.py +0 -0
  120. chialoops-1.0.0/chia/firesim/build_node.py +1942 -0
  121. chialoops-1.0.0/chia/firesim/chia_functions.py +162 -0
  122. chialoops-1.0.0/chia/firesim/config.py +334 -0
  123. chialoops-1.0.0/chia/firesim/run_node.py +788 -0
  124. chialoops-1.0.0/chia/firesim/spec_parser.py +131 -0
  125. chialoops-1.0.0/chia/firesim/state_def.py +81 -0
  126. chialoops-1.0.0/chia/firesim/suite_runner.py +477 -0
  127. chialoops-1.0.0/chia/firesim/test/test_firesim_build_e2e.py +96 -0
  128. chialoops-1.0.0/chia/firesim/test/test_firesim_build_megaboom_all.py +115 -0
  129. chialoops-1.0.0/chia/firesim/test/test_firesim_build_ray.py +129 -0
  130. chialoops-1.0.0/chia/firesim/test/test_firesim_metasim_build.py +84 -0
  131. chialoops-1.0.0/chia/firesim/test/test_firesim_run_e2e.py +116 -0
  132. chialoops-1.0.0/chia/firesim/test/test_firesim_suite_e2e.py +99 -0
  133. chialoops-1.0.0/chia/firesim/workloads.py +310 -0
  134. chialoops-1.0.0/chia/github/__init__.py +0 -0
  135. chialoops-1.0.0/chia/github/github_client.py +192 -0
  136. chialoops-1.0.0/chia/github/github_issues_node.py +226 -0
  137. chialoops-1.0.0/chia/github/github_pulls_node.py +269 -0
  138. chialoops-1.0.0/chia/github/state_def.py +205 -0
  139. chialoops-1.0.0/chia/github/test/test_github_issues_e2e.py +149 -0
  140. chialoops-1.0.0/chia/models/antigravity.py +757 -0
  141. chialoops-1.0.0/chia/models/bedrock.py +612 -0
  142. chialoops-1.0.0/chia/models/claude.py +1411 -0
  143. chialoops-1.0.0/chia/models/codex.py +738 -0
  144. chialoops-1.0.0/chia/models/copilot.py +734 -0
  145. chialoops-1.0.0/chia/models/ollama.py +60 -0
  146. chialoops-1.0.0/chia/models/openai_compat.py +728 -0
  147. chialoops-1.0.0/chia/models/openai_providers.py +97 -0
  148. chialoops-1.0.0/chia/models/opencode.py +889 -0
  149. chialoops-1.0.0/chia/models/tests/cluster/ollama_e2e_test.py +294 -0
  150. chialoops-1.0.0/chia/models/tests/cluster/ollama_tool_hammer.py +132 -0
  151. chialoops-1.0.0/chia/models/tests/cluster/vllm_e2e_test.py +196 -0
  152. chialoops-1.0.0/chia/models/tests/conftest.py +98 -0
  153. chialoops-1.0.0/chia/models/tests/fixtures/bedrock/capture_real_fixtures.py +68 -0
  154. chialoops-1.0.0/chia/models/tests/fixtures/make_synthetic_fixtures.py +87 -0
  155. chialoops-1.0.0/chia/models/tests/fixtures/vertex/capture_real_fixtures.py +72 -0
  156. chialoops-1.0.0/chia/models/tests/test_antigravity.py +477 -0
  157. chialoops-1.0.0/chia/models/tests/test_bedrock.py +678 -0
  158. chialoops-1.0.0/chia/models/tests/test_claude_api.py +1025 -0
  159. chialoops-1.0.0/chia/models/tests/test_codex.py +458 -0
  160. chialoops-1.0.0/chia/models/tests/test_copilot.py +530 -0
  161. chialoops-1.0.0/chia/models/tests/test_ollama.py +317 -0
  162. chialoops-1.0.0/chia/models/tests/test_openai_compat.py +996 -0
  163. chialoops-1.0.0/chia/models/tests/test_openai_providers.py +629 -0
  164. chialoops-1.0.0/chia/models/tests/test_opencode.py +1119 -0
  165. chialoops-1.0.0/chia/models/tests/test_vertex.py +908 -0
  166. chialoops-1.0.0/chia/models/tests/test_vllm.py +317 -0
  167. chialoops-1.0.0/chia/models/vertex.py +757 -0
  168. chialoops-1.0.0/chia/models/vllm.py +75 -0
  169. chialoops-1.0.0/chia/simulators/__init__.py +43 -0
  170. chialoops-1.0.0/chia/simulators/champsim.py +1152 -0
  171. chialoops-1.0.0/chia/simulators/gem5.py +1216 -0
  172. chialoops-1.0.0/chia/simulators/tests/__init__.py +0 -0
  173. chialoops-1.0.0/chia/simulators/tests/test_champsim_live.py +474 -0
  174. chialoops-1.0.0/chia/simulators/tests/test_champsim_smoke_trace.py +138 -0
  175. chialoops-1.0.0/chia/simulators/tests/test_gem5_live.py +658 -0
  176. chialoops-1.0.0/chia/simulators/tests/traces/generate_smoke_trace.py +66 -0
  177. chialoops-1.0.0/chia/trace/__init__.py +11 -0
  178. chialoops-1.0.0/chia/trace/metrics.py +158 -0
  179. chialoops-1.0.0/chia/trace/profile_table.py +163 -0
  180. chialoops-1.0.0/chia/trace/profile_template.html +777 -0
  181. chialoops-1.0.0/chia/trace/profile_viz.py +549 -0
  182. chialoops-1.0.0/chia/trace/profile_viz_html.py +181 -0
  183. chialoops-1.0.0/chia/trace/profiler.py +563 -0
  184. chialoops-1.0.0/chia/trace/test/test_metrics_logger.py +114 -0
  185. chialoops-1.0.0/chia/trace/test/test_profiler.py +597 -0
  186. chialoops-1.0.0/chia/trace/viz.py +733 -0
  187. chialoops-1.0.0/chia/vlsi/__init__.py +4 -0
  188. chialoops-1.0.0/chia/vlsi/hammer.py +396 -0
  189. chialoops-1.0.0/chia/vlsi/sram_cacti/__init__.py +0 -0
  190. chialoops-1.0.0/chia/vlsi/sram_cacti/cacti_macrocompiler.py +18 -0
  191. chialoops-1.0.0/chia/vlsi/sram_cacti/cacti_runner.py +218 -0
  192. chialoops-1.0.0/chia/vlsi/sram_cacti/cluster/cacti_smoke_test.py +360 -0
  193. chialoops-1.0.0/chia/vlsi/sram_cacti/lef_gen.py +189 -0
  194. chialoops-1.0.0/chia/vlsi/sram_cacti/liberty_gen.py +290 -0
  195. chialoops-1.0.0/chia/vlsi/sram_cacti/sram_characterize.py +217 -0
  196. chialoops-1.0.0/chia/vlsi/sram_cacti/test/conftest.py +80 -0
  197. chialoops-1.0.0/chia/vlsi/sram_cacti/test/test_cacti_macrocompiler.py +46 -0
  198. chialoops-1.0.0/chia/vlsi/sram_cacti/test/test_cacti_runner.py +122 -0
  199. chialoops-1.0.0/chia/vlsi/sram_cacti/test/test_file_validation.py +206 -0
  200. chialoops-1.0.0/chia/vlsi/sram_cacti/test/test_integration.py +71 -0
  201. chialoops-1.0.0/chia/vlsi/sram_cacti/test/test_lef_gen.py +64 -0
  202. chialoops-1.0.0/chia/vlsi/sram_cacti/test/test_liberty_gen.py +85 -0
  203. chialoops-1.0.0/chia/vlsi/sram_cacti/test/test_sram_characterize.py +129 -0
  204. chialoops-1.0.0/chia/vlsi/sram_cacti/test_nport.py +434 -0
  205. chialoops-1.0.0/chia/vlsi/tests/__init__.py +0 -0
  206. chialoops-1.0.0/chia/vlsi/tests/test_hammer_live.py +549 -0
  207. chialoops-1.0.0/chialoops.egg-info/PKG-INFO +149 -0
  208. chialoops-1.0.0/chialoops.egg-info/SOURCES.txt +212 -0
  209. chialoops-1.0.0/chialoops.egg-info/dependency_links.txt +1 -0
  210. chialoops-1.0.0/chialoops.egg-info/entry_points.txt +2 -0
  211. chialoops-1.0.0/chialoops.egg-info/requires.txt +35 -0
  212. chialoops-1.0.0/chialoops.egg-info/top_level.txt +1 -0
  213. chialoops-1.0.0/pyproject.toml +73 -0
  214. chialoops-1.0.0/setup.cfg +4 -0
@@ -0,0 +1,29 @@
1
+ BSD 3-Clause License
2
+
3
+ Copyright (c) 2026, The Regents of the University of California (Regents)
4
+ All Rights Reserved.
5
+
6
+ Redistribution and use in source and binary forms, with or without
7
+ modification, are permitted provided that the following conditions are met:
8
+
9
+ 1. Redistributions of source code must retain the above copyright notice, this
10
+ list of conditions and the following disclaimer.
11
+
12
+ 2. Redistributions in binary form must reproduce the above copyright notice,
13
+ this list of conditions and the following disclaimer in the documentation
14
+ and/or other materials provided with the distribution.
15
+
16
+ 3. Neither the name of the copyright holder nor the names of its
17
+ contributors may be used to endorse or promote products derived from
18
+ this software without specific prior written permission.
19
+
20
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
21
+ AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22
+ IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
23
+ DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
24
+ FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25
+ DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
26
+ SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
27
+ CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
28
+ OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29
+ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
@@ -0,0 +1,149 @@
1
+ Metadata-Version: 2.4
2
+ Name: chialoops
3
+ Version: 1.0.0
4
+ Summary: The Chia agentic AI flow for hardware
5
+ Author: UC Berkeley Architecture Research
6
+ License-Expression: BSD-3-Clause
7
+ Project-URL: Homepage, https://github.com/ucb-bar/chia
8
+ Project-URL: Documentation, https://docs.chialoops.ai
9
+ Project-URL: Repository, https://github.com/ucb-bar/chia
10
+ Project-URL: Issues, https://github.com/ucb-bar/chia/issues
11
+ Keywords: hardware,co-design,chipyard,agents,ray
12
+ Classifier: Development Status :: 5 - Production/Stable
13
+ Classifier: Intended Audience :: Science/Research
14
+ Classifier: Programming Language :: Python :: 3
15
+ Classifier: Topic :: Scientific/Engineering
16
+ Requires-Python: >=3.10
17
+ Description-Content-Type: text/markdown
18
+ License-File: LICENSE
19
+ Requires-Dist: google-genai>=1.64.0
20
+ Requires-Dist: ray[default]==2.54.0
21
+ Requires-Dist: mcp==1.27.1
22
+ Requires-Dist: pydantic==2.12.4
23
+ Requires-Dist: fastapi==0.121.0
24
+ Requires-Dist: pyyaml==6.0.3
25
+ Requires-Dist: graphviz==0.21
26
+ Requires-Dist: boto3>=1.28.0
27
+ Requires-Dist: google-cloud-compute>=1.19.0
28
+ Requires-Dist: requests>=2.31.0
29
+ Provides-Extra: test
30
+ Requires-Dist: pytest==9.0.3; extra == "test"
31
+ Provides-Extra: tensorboard
32
+ Requires-Dist: tensorboardX>=2.6; extra == "tensorboard"
33
+ Requires-Dist: tensorboard>=2.14; extra == "tensorboard"
34
+ Provides-Extra: wandb
35
+ Requires-Dist: wandb>=0.15; extra == "wandb"
36
+ Provides-Extra: metrics
37
+ Requires-Dist: tensorboardX>=2.6; extra == "metrics"
38
+ Requires-Dist: wandb>=0.15; extra == "metrics"
39
+ Provides-Extra: postgres
40
+ Requires-Dist: psycopg[binary]>=3.2; extra == "postgres"
41
+ Provides-Extra: docs
42
+ Requires-Dist: sphinx; extra == "docs"
43
+ Requires-Dist: sphinx_rtd_theme; extra == "docs"
44
+ Requires-Dist: sphinx-tabs; extra == "docs"
45
+ Requires-Dist: sphinx-copybutton; extra == "docs"
46
+ Requires-Dist: sphinx-substitution-extensions; extra == "docs"
47
+ Requires-Dist: sphinx-autobuild; extra == "docs"
48
+ Dynamic: license-file
49
+
50
+ <p align="center">
51
+ <picture>
52
+ <source media="(prefers-color-scheme: dark)" srcset="docs/_static/chia-logo-inv.png">
53
+ <img alt="CHIA" src="docs/_static/chia-logo.png" width="420"/>
54
+ </picture>
55
+ </p>
56
+
57
+ <h3 align="center">CHIA: An open-source framework for principled, agentic AI-driven hardware/software co-design research</h3>
58
+
59
+ <p align="center">
60
+ <a href="https://docs.chialoops.ai"><b>Documentation</b></a> &nbsp;•&nbsp;
61
+ <a href="https://openreview.net/pdf?id=lLxEUReWHG"><b>Paper</b></a> &nbsp;•&nbsp;
62
+ <a href="https://chialoops.ai"><b>Website</b></a>
63
+ </p>
64
+
65
+ ---
66
+
67
+ ## What is CHIA?
68
+
69
+ CHIA is an open-source framework for agile and principled hardware/software co-design research. Even though many of the steps of the hardware/software co-design process can be accelerated by AI, existing research using AI in these contexts has been limited to small studies on isolated examples because it is still too hard to assemble more complex experiments. CHIA solves this problem by enabling users to express the whole co-design **workflow** in an agile way with all of the tools you already use. CHIA abstracts workflows as graphs, and provides an efficient, feature rich runtime system to execute these workflows.
70
+
71
+ See the [documentation](https://docs.chialoops.ai/en/latest/getting-started/quickstart.html) to run your first flow!
72
+
73
+ ## Installation
74
+
75
+ CHIA requires **Python 3.10.19** (matching the Python in the Docker images). With conda:
76
+
77
+ ```bash
78
+ conda create -n chia_env python=3.10.19
79
+ conda activate chia_env
80
+ pip install -e /path/to/chia
81
+ ```
82
+
83
+ ## Learn more about CHIA
84
+
85
+ - **[CHIA Basics](https://docs.chialoops.ai/en/latest/getting-started/chia-basics.html)** — the core ideas, start here.
86
+ - **[Architecture Overview](https://docs.chialoops.ai/en/latest/concepts/overview.html)** — how CHIA works under the hood.
87
+
88
+ User guides:
89
+
90
+ - [ChiaFunction](https://docs.chialoops.ai/en/latest/user_guides/chia_function.html)
91
+ - [ChiaTool](https://docs.chialoops.ai/en/latest/user_guides/chia_tool.html)
92
+ - [Cluster Configuration Reference](https://docs.chialoops.ai/en/latest/user_guides/cluster_config_reference.html)
93
+ - [Building a CHIA-compatible Docker image](https://docs.chialoops.ai/en/latest/user_guides/docker_images.html)
94
+ - [Caching and Bypass](https://docs.chialoops.ai/en/latest/user_guides/caching_and_bypass.html)
95
+ - [Profiling](https://docs.chialoops.ai/en/latest/user_guides/profiling.html)
96
+
97
+ ## Overview of CHIA and Early Case Studies
98
+
99
+ <p align="center">
100
+ <img src="docs/_static/overviewfig.png" alt="CHIA overview" width="820"/>
101
+ </p>
102
+
103
+ ## Integrations & Ecosystem
104
+
105
+ CHIA workflows compose the tools you already use across the hardware/software co-design stack:
106
+
107
+ <p align="center">
108
+ <img src="docs/_static/project-logos/claude-code.svg" alt="Claude Code" height="40"/> &nbsp;&nbsp;
109
+ <img src="docs/_static/project-logos/openai-codex.svg" alt="OpenAI Codex" height="40"/> &nbsp;&nbsp;
110
+ <img src="docs/_static/project-logos/github-copilot.svg" alt="GitHub Copilot" height="40"/> &nbsp;&nbsp;
111
+ <img src="docs/_static/project-logos/google-antigravity.svg" alt="Google Antigravity" height="40"/> &nbsp;&nbsp;
112
+ <img src="docs/_static/project-logos/alphaevolve.svg" alt="AlphaEvolve" height="40"/> &nbsp;&nbsp;
113
+ <img src="docs/_static/project-logos/openevolve.png" alt="OpenEvolve" height="40"/> &nbsp;&nbsp;
114
+ <img src="docs/_static/project-logos/adaevolve.svg" alt="AdaEvolve" height="40"/> &nbsp;&nbsp;
115
+ <img src="docs/_static/project-logos/skydiscover.png" alt="SkyDiscover" height="40"/> &nbsp;&nbsp;
116
+ <img src="docs/_static/project-logos/chisel.svg" alt="Chisel" height="40"/> &nbsp;&nbsp;
117
+ <img src="docs/_static/project-logos/chipyard.svg" alt="Chipyard" height="40"/> &nbsp;&nbsp;
118
+ <img src="docs/_static/project-logos/circt.svg" alt="CIRCT" height="40"/> &nbsp;&nbsp;
119
+ <img src="docs/_static/project-logos/hammer.png" alt="Hammer" height="40"/> &nbsp;&nbsp;
120
+ <img src="docs/_static/project-logos/gem5.png" alt="gem5" height="40"/> &nbsp;&nbsp;
121
+ <img src="docs/_static/project-logos/firesim.png" alt="FireSim" height="40"/> &nbsp;&nbsp;
122
+ <img src="docs/_static/project-logos/verilator.png" alt="Verilator" height="40"/> &nbsp;&nbsp;
123
+ <img src="docs/_static/project-logos/spike.svg" alt="Spike" height="40"/> &nbsp;&nbsp;
124
+ <img src="docs/_static/project-logos/champsim.svg" alt="ChampSim" height="40"/> &nbsp;&nbsp;
125
+ </p>
126
+
127
+ <p align="center">
128
+ Powered by Ray:
129
+ </p>
130
+
131
+ <p align="center">
132
+ <img src="docs/_static/project-logos/ray.png" alt="Ray" height="40"/>
133
+ </p>
134
+
135
+ ## Attribution
136
+
137
+ If you use CHIA in your research, please cite our paper:
138
+
139
+ ```bibtex
140
+ @misc{cui2026chiaopensourceframeworkprincipled,
141
+ title={CHIA: An open-source framework for principled, agentic AI-driven hardware/software co-design research},
142
+ author={Angela Cui and Ferran Hermida-Rivera and Jack Toubes and Raghav Gupta and Jim Fang and Chengyi Lux Zhang and Ella Schwarz and Junha Kim and Yakun Sophia Shao and Borivoje Nikolic and Christopher W. Fletcher and Sagar Karandikar},
143
+ year={2026},
144
+ eprint={2606.27350},
145
+ archivePrefix={arXiv},
146
+ primaryClass={cs.AR},
147
+ url={https://arxiv.org/abs/2606.27350},
148
+ }
149
+ ```
@@ -0,0 +1,100 @@
1
+ <p align="center">
2
+ <picture>
3
+ <source media="(prefers-color-scheme: dark)" srcset="docs/_static/chia-logo-inv.png">
4
+ <img alt="CHIA" src="docs/_static/chia-logo.png" width="420"/>
5
+ </picture>
6
+ </p>
7
+
8
+ <h3 align="center">CHIA: An open-source framework for principled, agentic AI-driven hardware/software co-design research</h3>
9
+
10
+ <p align="center">
11
+ <a href="https://docs.chialoops.ai"><b>Documentation</b></a> &nbsp;•&nbsp;
12
+ <a href="https://openreview.net/pdf?id=lLxEUReWHG"><b>Paper</b></a> &nbsp;•&nbsp;
13
+ <a href="https://chialoops.ai"><b>Website</b></a>
14
+ </p>
15
+
16
+ ---
17
+
18
+ ## What is CHIA?
19
+
20
+ CHIA is an open-source framework for agile and principled hardware/software co-design research. Even though many of the steps of the hardware/software co-design process can be accelerated by AI, existing research using AI in these contexts has been limited to small studies on isolated examples because it is still too hard to assemble more complex experiments. CHIA solves this problem by enabling users to express the whole co-design **workflow** in an agile way with all of the tools you already use. CHIA abstracts workflows as graphs, and provides an efficient, feature rich runtime system to execute these workflows.
21
+
22
+ See the [documentation](https://docs.chialoops.ai/en/latest/getting-started/quickstart.html) to run your first flow!
23
+
24
+ ## Installation
25
+
26
+ CHIA requires **Python 3.10.19** (matching the Python in the Docker images). With conda:
27
+
28
+ ```bash
29
+ conda create -n chia_env python=3.10.19
30
+ conda activate chia_env
31
+ pip install -e /path/to/chia
32
+ ```
33
+
34
+ ## Learn more about CHIA
35
+
36
+ - **[CHIA Basics](https://docs.chialoops.ai/en/latest/getting-started/chia-basics.html)** — the core ideas, start here.
37
+ - **[Architecture Overview](https://docs.chialoops.ai/en/latest/concepts/overview.html)** — how CHIA works under the hood.
38
+
39
+ User guides:
40
+
41
+ - [ChiaFunction](https://docs.chialoops.ai/en/latest/user_guides/chia_function.html)
42
+ - [ChiaTool](https://docs.chialoops.ai/en/latest/user_guides/chia_tool.html)
43
+ - [Cluster Configuration Reference](https://docs.chialoops.ai/en/latest/user_guides/cluster_config_reference.html)
44
+ - [Building a CHIA-compatible Docker image](https://docs.chialoops.ai/en/latest/user_guides/docker_images.html)
45
+ - [Caching and Bypass](https://docs.chialoops.ai/en/latest/user_guides/caching_and_bypass.html)
46
+ - [Profiling](https://docs.chialoops.ai/en/latest/user_guides/profiling.html)
47
+
48
+ ## Overview of CHIA and Early Case Studies
49
+
50
+ <p align="center">
51
+ <img src="docs/_static/overviewfig.png" alt="CHIA overview" width="820"/>
52
+ </p>
53
+
54
+ ## Integrations & Ecosystem
55
+
56
+ CHIA workflows compose the tools you already use across the hardware/software co-design stack:
57
+
58
+ <p align="center">
59
+ <img src="docs/_static/project-logos/claude-code.svg" alt="Claude Code" height="40"/> &nbsp;&nbsp;
60
+ <img src="docs/_static/project-logos/openai-codex.svg" alt="OpenAI Codex" height="40"/> &nbsp;&nbsp;
61
+ <img src="docs/_static/project-logos/github-copilot.svg" alt="GitHub Copilot" height="40"/> &nbsp;&nbsp;
62
+ <img src="docs/_static/project-logos/google-antigravity.svg" alt="Google Antigravity" height="40"/> &nbsp;&nbsp;
63
+ <img src="docs/_static/project-logos/alphaevolve.svg" alt="AlphaEvolve" height="40"/> &nbsp;&nbsp;
64
+ <img src="docs/_static/project-logos/openevolve.png" alt="OpenEvolve" height="40"/> &nbsp;&nbsp;
65
+ <img src="docs/_static/project-logos/adaevolve.svg" alt="AdaEvolve" height="40"/> &nbsp;&nbsp;
66
+ <img src="docs/_static/project-logos/skydiscover.png" alt="SkyDiscover" height="40"/> &nbsp;&nbsp;
67
+ <img src="docs/_static/project-logos/chisel.svg" alt="Chisel" height="40"/> &nbsp;&nbsp;
68
+ <img src="docs/_static/project-logos/chipyard.svg" alt="Chipyard" height="40"/> &nbsp;&nbsp;
69
+ <img src="docs/_static/project-logos/circt.svg" alt="CIRCT" height="40"/> &nbsp;&nbsp;
70
+ <img src="docs/_static/project-logos/hammer.png" alt="Hammer" height="40"/> &nbsp;&nbsp;
71
+ <img src="docs/_static/project-logos/gem5.png" alt="gem5" height="40"/> &nbsp;&nbsp;
72
+ <img src="docs/_static/project-logos/firesim.png" alt="FireSim" height="40"/> &nbsp;&nbsp;
73
+ <img src="docs/_static/project-logos/verilator.png" alt="Verilator" height="40"/> &nbsp;&nbsp;
74
+ <img src="docs/_static/project-logos/spike.svg" alt="Spike" height="40"/> &nbsp;&nbsp;
75
+ <img src="docs/_static/project-logos/champsim.svg" alt="ChampSim" height="40"/> &nbsp;&nbsp;
76
+ </p>
77
+
78
+ <p align="center">
79
+ Powered by Ray:
80
+ </p>
81
+
82
+ <p align="center">
83
+ <img src="docs/_static/project-logos/ray.png" alt="Ray" height="40"/>
84
+ </p>
85
+
86
+ ## Attribution
87
+
88
+ If you use CHIA in your research, please cite our paper:
89
+
90
+ ```bibtex
91
+ @misc{cui2026chiaopensourceframeworkprincipled,
92
+ title={CHIA: An open-source framework for principled, agentic AI-driven hardware/software co-design research},
93
+ author={Angela Cui and Ferran Hermida-Rivera and Jack Toubes and Raghav Gupta and Jim Fang and Chengyi Lux Zhang and Ella Schwarz and Junha Kim and Yakun Sophia Shao and Borivoje Nikolic and Christopher W. Fletcher and Sagar Karandikar},
94
+ year={2026},
95
+ eprint={2606.27350},
96
+ archivePrefix={arXiv},
97
+ primaryClass={cs.AR},
98
+ url={https://arxiv.org/abs/2606.27350},
99
+ }
100
+ ```
@@ -0,0 +1 @@
1
+ """Post-design analysis utilities (carbon/sustainability modeling, etc.)."""
@@ -0,0 +1,23 @@
1
+ from chia.analysis.sustainability.focal import (
2
+ ALPHA_EMBODIED_DOMINATED,
3
+ ALPHA_OPERATIONAL_DOMINATED,
4
+ SWEEPABLE_PARAMETERS,
5
+ FocalBothRefs,
6
+ FocalComparison,
7
+ FocalSweep,
8
+ focal_compare,
9
+ focal_sweep,
10
+ sweep_both_arches_as_ref,
11
+ )
12
+
13
+ __all__ = [
14
+ "ALPHA_EMBODIED_DOMINATED",
15
+ "ALPHA_OPERATIONAL_DOMINATED",
16
+ "SWEEPABLE_PARAMETERS",
17
+ "FocalBothRefs",
18
+ "FocalComparison",
19
+ "FocalSweep",
20
+ "focal_compare",
21
+ "focal_sweep",
22
+ "sweep_both_arches_as_ref",
23
+ ]