pantheon-gpu 1.2.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 (142) hide show
  1. pantheon_gpu-1.2.0/LICENSE +201 -0
  2. pantheon_gpu-1.2.0/NOTICE +42 -0
  3. pantheon_gpu-1.2.0/PKG-INFO +270 -0
  4. pantheon_gpu-1.2.0/README.md +251 -0
  5. pantheon_gpu-1.2.0/pantheon_gpu/Makefile +207 -0
  6. pantheon_gpu-1.2.0/pantheon_gpu/VERSION +1 -0
  7. pantheon_gpu-1.2.0/pantheon_gpu/__init__.py +1 -0
  8. pantheon_gpu-1.2.0/pantheon_gpu/_entry.py +17 -0
  9. pantheon_gpu-1.2.0/pantheon_gpu/kernels/all_reduce/README.md +17 -0
  10. pantheon_gpu-1.2.0/pantheon_gpu/kernels/all_reduce/all_reduce.cpp +178 -0
  11. pantheon_gpu-1.2.0/pantheon_gpu/kernels/allocation_fragmentation/README.md +3 -0
  12. pantheon_gpu-1.2.0/pantheon_gpu/kernels/allocation_fragmentation/allocation_fragmentation.cpp +177 -0
  13. pantheon_gpu-1.2.0/pantheon_gpu/kernels/atomic_virus/README.md +57 -0
  14. pantheon_gpu-1.2.0/pantheon_gpu/kernels/atomic_virus/atomic_virus.cpp +266 -0
  15. pantheon_gpu-1.2.0/pantheon_gpu/kernels/atomic_virus/atomic_virus_flow.svg +33 -0
  16. pantheon_gpu-1.2.0/pantheon_gpu/kernels/cache_latency/README.md +57 -0
  17. pantheon_gpu-1.2.0/pantheon_gpu/kernels/cache_latency/cache_latency.cpp +279 -0
  18. pantheon_gpu-1.2.0/pantheon_gpu/kernels/cache_latency/cache_latency_flow.svg +30 -0
  19. pantheon_gpu-1.2.0/pantheon_gpu/kernels/common/ai_workload_template.h +366 -0
  20. pantheon_gpu-1.2.0/pantheon_gpu/kernels/common/common.h +427 -0
  21. pantheon_gpu-1.2.0/pantheon_gpu/kernels/common/fault_log.h +136 -0
  22. pantheon_gpu-1.2.0/pantheon_gpu/kernels/common/fp16_shim.h +20 -0
  23. pantheon_gpu-1.2.0/pantheon_gpu/kernels/common/mock_gpu.h +212 -0
  24. pantheon_gpu-1.2.0/pantheon_gpu/kernels/common/nvenc/nvEncodeAPI.h +4667 -0
  25. pantheon_gpu-1.2.0/pantheon_gpu/kernels/compute_virus/README.md +59 -0
  26. pantheon_gpu-1.2.0/pantheon_gpu/kernels/compute_virus/compute_virus.cpp +223 -0
  27. pantheon_gpu-1.2.0/pantheon_gpu/kernels/compute_virus/compute_virus_agg.cpp +280 -0
  28. pantheon_gpu-1.2.0/pantheon_gpu/kernels/compute_virus/compute_virus_flow.svg +36 -0
  29. pantheon_gpu-1.2.0/pantheon_gpu/kernels/fp64_virus/README.md +96 -0
  30. pantheon_gpu-1.2.0/pantheon_gpu/kernels/fp64_virus/fp64_virus.cpp +249 -0
  31. pantheon_gpu-1.2.0/pantheon_gpu/kernels/fp64_virus/fp64_virus_flow.svg +95 -0
  32. pantheon_gpu-1.2.0/pantheon_gpu/kernels/fused_attention/README.md +3 -0
  33. pantheon_gpu-1.2.0/pantheon_gpu/kernels/fused_attention/fused_attention.cpp +2 -0
  34. pantheon_gpu-1.2.0/pantheon_gpu/kernels/galpat/README.md +62 -0
  35. pantheon_gpu-1.2.0/pantheon_gpu/kernels/galpat/galpat.cpp +233 -0
  36. pantheon_gpu-1.2.0/pantheon_gpu/kernels/graph_replay/README.md +3 -0
  37. pantheon_gpu-1.2.0/pantheon_gpu/kernels/graph_replay/graph_replay.cpp +173 -0
  38. pantheon_gpu-1.2.0/pantheon_gpu/kernels/idle/README.md +34 -0
  39. pantheon_gpu-1.2.0/pantheon_gpu/kernels/idle/idle.cpp +29 -0
  40. pantheon_gpu-1.2.0/pantheon_gpu/kernels/idle/idle_flow.svg +11 -0
  41. pantheon_gpu-1.2.0/pantheon_gpu/kernels/int_virus/README.md +57 -0
  42. pantheon_gpu-1.2.0/pantheon_gpu/kernels/int_virus/int_virus.cpp +272 -0
  43. pantheon_gpu-1.2.0/pantheon_gpu/kernels/int_virus/int_virus_flow.svg +32 -0
  44. pantheon_gpu-1.2.0/pantheon_gpu/kernels/kv_cache_churn/README.md +20 -0
  45. pantheon_gpu-1.2.0/pantheon_gpu/kernels/kv_cache_churn/kv_cache_churn.cpp +37 -0
  46. pantheon_gpu-1.2.0/pantheon_gpu/kernels/llm_decode/README.md +25 -0
  47. pantheon_gpu-1.2.0/pantheon_gpu/kernels/llm_decode/llm_decode.cpp +113 -0
  48. pantheon_gpu-1.2.0/pantheon_gpu/kernels/llm_prefill/README.md +22 -0
  49. pantheon_gpu-1.2.0/pantheon_gpu/kernels/llm_prefill/llm_prefill.cpp +55 -0
  50. pantheon_gpu-1.2.0/pantheon_gpu/kernels/march_test/README.md +72 -0
  51. pantheon_gpu-1.2.0/pantheon_gpu/kernels/march_test/march_test.cpp +243 -0
  52. pantheon_gpu-1.2.0/pantheon_gpu/kernels/media_enc_virus/README.md +38 -0
  53. pantheon_gpu-1.2.0/pantheon_gpu/kernels/media_enc_virus/media_enc_virus.cpp +267 -0
  54. pantheon_gpu-1.2.0/pantheon_gpu/kernels/media_enc_virus/media_enc_virus_flow.svg +10 -0
  55. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_bank_thrash/README.md +42 -0
  56. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_bank_thrash/memory_bank_thrash.cpp +228 -0
  57. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_bank_thrash/memory_bank_thrash_flow.svg +10 -0
  58. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_cache_fracture/README.md +42 -0
  59. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_cache_fracture/memory_cache_fracture.cpp +223 -0
  60. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_cache_fracture/memory_cache_fracture_flow.svg +9 -0
  61. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_hammer/README.md +83 -0
  62. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_hammer/memory_hammer.cpp +299 -0
  63. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_pc_pingpong/README.md +42 -0
  64. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_pc_pingpong/memory_pc_pingpong.cpp +248 -0
  65. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_pc_pingpong/memory_pc_pingpong_flow.svg +9 -0
  66. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_read/README.md +69 -0
  67. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_read/memory_read.cpp +295 -0
  68. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_retention/README.md +67 -0
  69. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_retention/memory_retention.cpp +193 -0
  70. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_retention_bake/README.md +41 -0
  71. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_retention_bake/memory_retention_bake.cpp +245 -0
  72. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_retention_bake/memory_retention_bake_flow.svg +10 -0
  73. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_thermal_asym/README.md +41 -0
  74. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_thermal_asym/memory_thermal_asym.cpp +230 -0
  75. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_thermal_asym/memory_thermal_asym_flow.svg +7 -0
  76. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_tsv_thrasher/README.md +42 -0
  77. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_tsv_thrasher/memory_tsv_thrasher.cpp +242 -0
  78. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_tsv_thrasher/memory_tsv_thrasher_flow.svg +8 -0
  79. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_write/README.md +71 -0
  80. pantheon_gpu-1.2.0/pantheon_gpu/kernels/memory_write/memory_write.cpp +262 -0
  81. pantheon_gpu-1.2.0/pantheon_gpu/kernels/mma_virus/README.md +42 -0
  82. pantheon_gpu-1.2.0/pantheon_gpu/kernels/mma_virus/mma_virus.cpp +288 -0
  83. pantheon_gpu-1.2.0/pantheon_gpu/kernels/mma_virus/mma_virus_flow.svg +10 -0
  84. pantheon_gpu-1.2.0/pantheon_gpu/kernels/moe_router/README.md +3 -0
  85. pantheon_gpu-1.2.0/pantheon_gpu/kernels/moe_router/moe_router.cpp +2 -0
  86. pantheon_gpu-1.2.0/pantheon_gpu/kernels/omni_virus/README.md +43 -0
  87. pantheon_gpu-1.2.0/pantheon_gpu/kernels/omni_virus/omni_virus.cpp +375 -0
  88. pantheon_gpu-1.2.0/pantheon_gpu/kernels/omni_virus/omni_virus_flow.svg +7 -0
  89. pantheon_gpu-1.2.0/pantheon_gpu/kernels/p2p_thrasher/README.md +57 -0
  90. pantheon_gpu-1.2.0/pantheon_gpu/kernels/p2p_thrasher/p2p_thrasher.cpp +280 -0
  91. pantheon_gpu-1.2.0/pantheon_gpu/kernels/p2p_thrasher/p2p_thrasher_flow.svg +27 -0
  92. pantheon_gpu-1.2.0/pantheon_gpu/kernels/pcie_bandwidth/README.md +56 -0
  93. pantheon_gpu-1.2.0/pantheon_gpu/kernels/pcie_bandwidth/pcie_bandwidth.cpp +284 -0
  94. pantheon_gpu-1.2.0/pantheon_gpu/kernels/pcie_bandwidth/pcie_bandwidth_flow.svg +19 -0
  95. pantheon_gpu-1.2.0/pantheon_gpu/kernels/pulse_virus/README.md +56 -0
  96. pantheon_gpu-1.2.0/pantheon_gpu/kernels/pulse_virus/pulse_virus.cpp +256 -0
  97. pantheon_gpu-1.2.0/pantheon_gpu/kernels/pulse_virus/pulse_virus_flow.svg +18 -0
  98. pantheon_gpu-1.2.0/pantheon_gpu/kernels/quantized_gemm/README.md +3 -0
  99. pantheon_gpu-1.2.0/pantheon_gpu/kernels/quantized_gemm/quantized_gemm.cpp +2 -0
  100. pantheon_gpu-1.2.0/pantheon_gpu/kernels/rag_embedding/README.md +3 -0
  101. pantheon_gpu-1.2.0/pantheon_gpu/kernels/rag_embedding/rag_embedding.cpp +2 -0
  102. pantheon_gpu-1.2.0/pantheon_gpu/kernels/ras_validator/README.md +41 -0
  103. pantheon_gpu-1.2.0/pantheon_gpu/kernels/ras_validator/ras_validator.cpp +270 -0
  104. pantheon_gpu-1.2.0/pantheon_gpu/kernels/ras_validator/ras_validator_flow.svg +10 -0
  105. pantheon_gpu-1.2.0/pantheon_gpu/kernels/rope_stress/README.md +3 -0
  106. pantheon_gpu-1.2.0/pantheon_gpu/kernels/rope_stress/rope_stress.cpp +2 -0
  107. pantheon_gpu-1.2.0/pantheon_gpu/kernels/rt_virus/README.md +56 -0
  108. pantheon_gpu-1.2.0/pantheon_gpu/kernels/rt_virus/rt_virus.cpp +567 -0
  109. pantheon_gpu-1.2.0/pantheon_gpu/kernels/rt_virus/rt_virus_flow.svg +20 -0
  110. pantheon_gpu-1.2.0/pantheon_gpu/kernels/scheduler_virus/README.md +42 -0
  111. pantheon_gpu-1.2.0/pantheon_gpu/kernels/scheduler_virus/scheduler_virus.cpp +260 -0
  112. pantheon_gpu-1.2.0/pantheon_gpu/kernels/scheduler_virus/scheduler_virus_flow.svg +9 -0
  113. pantheon_gpu-1.2.0/pantheon_gpu/kernels/serving_mix/README.md +3 -0
  114. pantheon_gpu-1.2.0/pantheon_gpu/kernels/serving_mix/serving_mix.cpp +2 -0
  115. pantheon_gpu-1.2.0/pantheon_gpu/kernels/sfu_stress/README.md +57 -0
  116. pantheon_gpu-1.2.0/pantheon_gpu/kernels/sfu_stress/sfu_stress.cpp +246 -0
  117. pantheon_gpu-1.2.0/pantheon_gpu/kernels/sfu_stress/sfu_stress_flow.svg +31 -0
  118. pantheon_gpu-1.2.0/pantheon_gpu/kernels/speculative_decode/README.md +3 -0
  119. pantheon_gpu-1.2.0/pantheon_gpu/kernels/speculative_decode/speculative_decode.cpp +2 -0
  120. pantheon_gpu-1.2.0/pantheon_gpu/kernels/tensor_virus/README.md +56 -0
  121. pantheon_gpu-1.2.0/pantheon_gpu/kernels/tensor_virus/tensor_virus.cpp +251 -0
  122. pantheon_gpu-1.2.0/pantheon_gpu/kernels/tensor_virus/tensor_virus_flow.svg +29 -0
  123. pantheon_gpu-1.2.0/pantheon_gpu/kernels/tlb_avalanche/README.md +57 -0
  124. pantheon_gpu-1.2.0/pantheon_gpu/kernels/tlb_avalanche/tlb_avalanche.cpp +295 -0
  125. pantheon_gpu-1.2.0/pantheon_gpu/kernels/tlb_avalanche/tlb_avalanche_flow.svg +35 -0
  126. pantheon_gpu-1.2.0/pantheon_gpu/kernels/transformer_train_step/README.md +3 -0
  127. pantheon_gpu-1.2.0/pantheon_gpu/kernels/transformer_train_step/transformer_train_step.cpp +2 -0
  128. pantheon_gpu-1.2.0/pantheon_gpu/kernels/transformer_virus/README.md +43 -0
  129. pantheon_gpu-1.2.0/pantheon_gpu/kernels/transformer_virus/transformer_virus.cpp +393 -0
  130. pantheon_gpu-1.2.0/pantheon_gpu/kernels/transformer_virus/transformer_virus_flow.svg +10 -0
  131. pantheon_gpu-1.2.0/pantheon_gpu/kernels/vision_encoder/README.md +3 -0
  132. pantheon_gpu-1.2.0/pantheon_gpu/kernels/vision_encoder/vision_encoder.cpp +2 -0
  133. pantheon_gpu-1.2.0/pantheon_gpu/monitor.py +417 -0
  134. pantheon_gpu-1.2.0/pantheon_gpu/pantheon.py +2572 -0
  135. pantheon_gpu-1.2.0/pantheon_gpu.egg-info/PKG-INFO +270 -0
  136. pantheon_gpu-1.2.0/pantheon_gpu.egg-info/SOURCES.txt +140 -0
  137. pantheon_gpu-1.2.0/pantheon_gpu.egg-info/dependency_links.txt +1 -0
  138. pantheon_gpu-1.2.0/pantheon_gpu.egg-info/entry_points.txt +3 -0
  139. pantheon_gpu-1.2.0/pantheon_gpu.egg-info/requires.txt +7 -0
  140. pantheon_gpu-1.2.0/pantheon_gpu.egg-info/top_level.txt +1 -0
  141. pantheon_gpu-1.2.0/pyproject.toml +44 -0
  142. pantheon_gpu-1.2.0/setup.cfg +4 -0
@@ -0,0 +1,201 @@
1
+ Apache License
2
+ Version 2.0, January 2004
3
+ http://www.apache.org/licenses/
4
+
5
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
6
+
7
+ 1. Definitions.
8
+
9
+ "License" shall mean the terms and conditions for use, reproduction,
10
+ and distribution as defined by Sections 1 through 9 of this document.
11
+
12
+ "Licensor" shall mean the copyright owner or entity authorized by
13
+ the copyright owner that is granting the License.
14
+
15
+ "Legal Entity" shall mean the union of the acting entity and all
16
+ other entities that control, are controlled by, or are under common
17
+ control with that entity. For the purposes of this definition,
18
+ "control" means (i) the power, direct or indirect, to cause the
19
+ direction or management of such entity, whether by contract or
20
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
21
+ outstanding shares, or (iii) beneficial ownership of such entity.
22
+
23
+ "You" (or "Your") shall mean an individual or Legal Entity
24
+ exercising permissions granted by this License.
25
+
26
+ "Source" form shall mean the preferred form for making modifications,
27
+ including but not limited to software source code, documentation
28
+ source, and configuration files.
29
+
30
+ "Object" form shall mean any form resulting from mechanical
31
+ transformation or translation of a Source form, including but
32
+ not limited to compiled object code, generated documentation,
33
+ and conversions to other media types.
34
+
35
+ "Work" shall mean the work of authorship, whether in Source or
36
+ Object form, made available under the License, as indicated by a
37
+ copyright notice that is included in or attached to the work
38
+ (an example is provided in the Appendix below).
39
+
40
+ "Derivative Works" shall mean any work, whether in Source or Object
41
+ form, that is based on (or derived from) the Work and for which the
42
+ editorial revisions, annotations, elaborations, or other modifications
43
+ represent, as a whole, an original work of authorship. For the purposes
44
+ of this License, Derivative Works shall not include works that remain
45
+ separable from, or merely link (or bind by name) to the interfaces of,
46
+ the Work and Derivative Works thereof.
47
+
48
+ "Contribution" shall mean any work of authorship, including
49
+ the original version of the Work and any modifications or additions
50
+ to that Work or Derivative Works thereof, that is intentionally
51
+ submitted to Licensor for inclusion in the Work by the copyright owner
52
+ or by an individual or Legal Entity authorized to submit on behalf of
53
+ the copyright owner. For the purposes of this definition, "submitted"
54
+ means any form of electronic, verbal, or written communication sent
55
+ to the Licensor or its representatives, including but not limited to
56
+ communication on electronic mailing lists, source code control systems,
57
+ and issue tracking systems that are managed by, or on behalf of, the
58
+ Licensor for the purpose of discussing and improving the Work, but
59
+ excluding communication that is conspicuously marked or otherwise
60
+ designated in writing by the copyright owner as "Not a Contribution."
61
+
62
+ "Contributor" shall mean Licensor and any individual or Legal Entity
63
+ on behalf of whom a Contribution has been received by Licensor and
64
+ subsequently incorporated within the Work.
65
+
66
+ 2. Grant of Copyright License. Subject to the terms and conditions of
67
+ this License, each Contributor hereby grants to You a perpetual,
68
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
69
+ copyright license to reproduce, prepare Derivative Works of,
70
+ publicly display, publicly perform, sublicense, and distribute the
71
+ Work and such Derivative Works in Source or Object form.
72
+
73
+ 3. Grant of Patent License. Subject to the terms and conditions of
74
+ this License, each Contributor hereby grants to You a perpetual,
75
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
76
+ (except as stated in this section) patent license to make, have made,
77
+ use, offer to sell, sell, import, and otherwise transfer the Work,
78
+ where such license applies only to those patent claims licensable
79
+ by such Contributor that are necessarily infringed by their
80
+ Contribution(s) alone or by combination of their Contribution(s)
81
+ with the Work to which such Contribution(s) was submitted. If You
82
+ institute patent litigation against any entity (including a
83
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
84
+ or a Contribution incorporated within the Work constitutes direct
85
+ or contributory patent infringement, then any patent licenses
86
+ granted to You under this License for that Work shall terminate
87
+ as of the date such litigation is filed.
88
+
89
+ 4. Redistribution. You may reproduce and distribute copies of the
90
+ Work or Derivative Works thereof in any medium, with or without
91
+ modifications, and in Source or Object form, provided that You
92
+ meet the following conditions:
93
+
94
+ (a) You must give any other recipients of the Work or
95
+ Derivative Works a copy of this License; and
96
+
97
+ (b) You must cause any modified files to carry prominent notices
98
+ stating that You changed the files; and
99
+
100
+ (c) You must retain, in the Source form of any Derivative Works
101
+ that You distribute, all copyright, patent, trademark, and
102
+ attribution notices from the Source form of the Work,
103
+ excluding those notices that do not pertain to any part of
104
+ the Derivative Works; and
105
+
106
+ (d) If the Work includes a "NOTICE" text file as part of its
107
+ distribution, then any Derivative Works that You distribute must
108
+ include a readable copy of the attribution notices contained
109
+ within such NOTICE file, excluding those notices that do not
110
+ pertain to any part of the Derivative Works, in at least one
111
+ of the following places: within a NOTICE text file distributed
112
+ as part of the Derivative Works; within the Source form or
113
+ documentation, if provided along with the Derivative Works; or,
114
+ within a display generated by the Derivative Works, if and
115
+ wherever such third-party notices normally appear. The contents
116
+ of the NOTICE file are for informational purposes only and
117
+ do not modify the License. You may add Your own attribution
118
+ notices within Derivative Works that You distribute, alongside
119
+ or as an addendum to the NOTICE text from the Work, provided
120
+ that such additional attribution notices cannot be construed
121
+ as modifying the License.
122
+
123
+ You may add Your own copyright statement to Your modifications and
124
+ may provide additional or different license terms and conditions
125
+ for use, reproduction, or distribution of Your modifications, or
126
+ for any such Derivative Works as a whole, provided Your use,
127
+ reproduction, and distribution of the Work otherwise complies with
128
+ the conditions stated in this License.
129
+
130
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
131
+ any Contribution intentionally submitted for inclusion in the Work
132
+ by You to the Licensor shall be under the terms and conditions of
133
+ this License, without any additional terms or conditions.
134
+ Notwithstanding the above, nothing herein shall supersede or modify
135
+ the terms of any separate license agreement you may have executed
136
+ with Licensor regarding such Contributions.
137
+
138
+ 6. Trademarks. This License does not grant permission to use the trade
139
+ names, trademarks, service marks, or product names of the Licensor,
140
+ except as required for reasonable and customary use in describing the
141
+ origin of the Work and reproducing the content of the NOTICE file.
142
+
143
+ 7. Disclaimer of Warranty. Unless required by applicable law or
144
+ agreed to in writing, Licensor provides the Work (and each
145
+ Contributor provides its Contributions) on an "AS IS" BASIS,
146
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
147
+ implied, including, without limitation, any warranties or conditions
148
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
149
+ PARTICULAR PURPOSE. You are solely responsible for determining the
150
+ appropriateness of using or redistributing the Work and assume any
151
+ risks associated with Your exercise of permissions under this License.
152
+
153
+ 8. Limitation of Liability. In no event and under no legal theory,
154
+ whether in tort (including negligence), contract, or otherwise,
155
+ unless required by applicable law (such as deliberate and grossly
156
+ negligent acts) or agreed to in writing, shall any Contributor be
157
+ liable to You for damages, including any direct, indirect, special,
158
+ incidental, or consequential damages of any character arising as a
159
+ result of this License or out of the use or inability to use the
160
+ Work (including but not limited to damages for loss of goodwill,
161
+ work stoppage, computer failure or malfunction, or any and all
162
+ other commercial damages or losses), even if such Contributor
163
+ has been advised of the possibility of such damages.
164
+
165
+ 9. Accepting Warranty or Additional Liability. While redistributing
166
+ the Work or Derivative Works thereof, You may choose to offer,
167
+ and charge a fee for, acceptance of support, warranty, indemnity,
168
+ or other liability obligations and/or rights consistent with this
169
+ License. However, in accepting such obligations, You may act only
170
+ on Your own behalf and on Your sole responsibility, not on behalf
171
+ of any other Contributor, and only if You agree to indemnify,
172
+ defend, and hold each Contributor harmless for any liability
173
+ incurred by, or claims asserted against, such Contributor by reason
174
+ of your accepting any such warranty or additional liability.
175
+
176
+ END OF TERMS AND CONDITIONS
177
+
178
+ APPENDIX: How to apply the Apache License to your work.
179
+
180
+ To apply the Apache License to your work, attach the following
181
+ boilerplate notice, with the fields enclosed by brackets "[]"
182
+ replaced with your own identifying information. (Don't include
183
+ the brackets!) The text should be enclosed in the appropriate
184
+ comment syntax for the file format. We also recommend that a
185
+ file or class name and description of purpose be included on the
186
+ same "printed page" as the copyright notice for easier
187
+ identification within third-party archives.
188
+
189
+ Copyright [yyyy] [name of copyright owner]
190
+
191
+ Licensed under the Apache License, Version 2.0 (the "License");
192
+ you may not use this file except in compliance with the License.
193
+ You may obtain a copy of the License at
194
+
195
+ http://www.apache.org/licenses/LICENSE-2.0
196
+
197
+ Unless required by applicable law or agreed to in writing, software
198
+ distributed under the License is distributed on an "AS IS" BASIS,
199
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
200
+ See the License for the specific language governing permissions and
201
+ limitations under the License.
@@ -0,0 +1,42 @@
1
+ Pantheon GPU
2
+ Copyright (c) Pantheon GPU contributors
3
+
4
+ This product is licensed under the Apache License, Version 2.0. See LICENSE.
5
+
6
+ ================================================================================
7
+ Third-party components bundled in this repository
8
+ ================================================================================
9
+
10
+ kernels/common/nvenc/nvEncodeAPI.h
11
+ Copyright (c) 2010-2024 NVIDIA Corporation.
12
+ Licensed under MIT-style terms that expressly permit use, copying,
13
+ modification, publication and distribution, provided the copyright notice
14
+ and permission notice are retained. The notice is preserved in the file.
15
+ Redistributable: yes.
16
+
17
+ ================================================================================
18
+ Third-party components required but NOT bundled
19
+ ================================================================================
20
+
21
+ NVIDIA OptiX SDK headers (optix.h, optix_stubs.h and friends)
22
+ Copyright (c) NVIDIA Corporation.
23
+ These headers state that use, reproduction, disclosure or distribution
24
+ without an express license agreement from NVIDIA is strictly prohibited.
25
+ They are therefore NOT redistributed with Pantheon.
26
+
27
+ Only the rt_virus workload needs them. The build probes for them and, when
28
+ they are absent, rt_virus builds a dummy kernel and reports its run as
29
+ skipped rather than failing the build. To build the real RT kernel, obtain
30
+ the OptiX SDK from NVIDIA and point the build at its include directory:
31
+
32
+ make PLATFORM=CUDA OPTIX_PATH=/path/to/optix/include
33
+
34
+ AMD HIP-RT headers (hiprt/hiprt.h)
35
+ Copyright (c) Advanced Micro Devices, Inc.
36
+ Not bundled. Handled the same way: rt_virus builds its dummy kernel when
37
+ they are absent. See the HIP-RT vendoring target in the Makefile.
38
+
39
+ ================================================================================
40
+
41
+ If you add a third-party file to this repository, record it above and confirm
42
+ its license permits redistribution first.
@@ -0,0 +1,270 @@
1
+ Metadata-Version: 2.4
2
+ Name: pantheon-gpu
3
+ Version: 1.2.0
4
+ Summary: GPU stress testing and diagnostics for NVIDIA CUDA and AMD ROCm
5
+ License: Apache-2.0
6
+ Project-URL: Homepage, https://pantheongpu.com
7
+ Project-URL: Source, https://github.com/pantheongpu/pantheon
8
+ Requires-Python: >=3.9
9
+ Description-Content-Type: text/markdown
10
+ License-File: LICENSE
11
+ License-File: NOTICE
12
+ Requires-Dist: psutil
13
+ Requires-Dist: pandas
14
+ Requires-Dist: numpy
15
+ Requires-Dist: nvidia-ml-py
16
+ Provides-Extra: reports
17
+ Requires-Dist: openpyxl; extra == "reports"
18
+ Dynamic: license-file
19
+
20
+ # Pantheon: Universal GPU Stress & Diagnostics Suite
21
+
22
+ Pantheon is a cross-platform (CUDA/ROCm) stress testing tool designed to isolate and hammer specific GPU subsystems. Unlike generic benchmarks (Furmark, 3DMark), Pantheon allows you to test specific silicon limits.
23
+
24
+ ## Requirements
25
+
26
+ ### Running From Source
27
+
28
+ This repository is the source tree. It does not build installable packages or
29
+ a standalone executable; run the checked-out runner directly:
30
+
31
+ ```bash
32
+ python3 pantheon.py --test baseline_metrics --duration 10
33
+ python3 pantheon.py --test tensor_virus --duration 60 --gpu 0
34
+ ```
35
+
36
+ Pantheon auto-detects CUDA, ROCm/HIP, or mock mode at runtime. Do not pass
37
+ `--platform cuda` for normal use.
38
+
39
+ Kernels are compiled on first use, or explicitly:
40
+
41
+ ```bash
42
+ make PLATFORM=CUDA # NVIDIA
43
+ make PLATFORM=HIP # AMD
44
+ make PLATFORM=MOCK # CPU backend, no GPU required
45
+ ```
46
+
47
+ ### Runtime Dependencies
48
+ - **Hardware Drivers:** Latest NVIDIA or AMD Linux drivers for the target GPU.
49
+ - **Compiler:** CUDA `nvcc` for NVIDIA GPUs, or ROCm/HIP `hipcc` for AMD GPUs. Pantheon uses the local compiler to build GPU workload binaries for the current machine.
50
+ - **System tools:** `make` and a C++ compiler such as `g++`.
51
+ - **Linux compatibility:** Official x86-64 binaries are built in a `manylinux_2_28` environment and require GLIBC 2.28 or newer.
52
+
53
+ ### Compiler Requirements
54
+ To run stress tests natively (and avoid **Mock Mode**), you must have the backend compiler for your specific GPU architecture installed:
55
+
56
+ #### For NVIDIA GPUs (CUDA)
57
+ **Linux (Ubuntu/Debian):**
58
+ ```bash
59
+ sudo apt-get update
60
+ sudo apt-get install nvidia-cuda-toolkit
61
+ ```
62
+
63
+ #### For AMD GPUs (ROCm/HIP)
64
+ **Linux (Ubuntu/Debian):**
65
+ ```bash
66
+ sudo apt-get update
67
+ sudo apt-get install hipcc
68
+ ```
69
+
70
+ ## Quick Start
71
+ ```bash
72
+ python3 pantheon.py --version
73
+
74
+ # Run a specific "Power Virus" test
75
+ python3 pantheon.py --test tensor_virus --duration 60
76
+
77
+ # Run a specific "VRM Cracker" test
78
+ python3 pantheon.py --test pulse_virus --duration 60
79
+
80
+ # Run the inference-path stress suite on GPU 0
81
+ python3 pantheon.py --test inference --duration 60 --gpu 0 --mem 50
82
+ ```
83
+
84
+ ## Ray tracing workload (rt_virus)
85
+
86
+ NVIDIA's OptiX headers are proprietary and not redistributable, so this tree
87
+ does not ship them and `rt_virus` builds a dummy kernel that reports
88
+ `SKIP` by default. To run the real workload on NVIDIA hardware, download the
89
+ [OptiX SDK](https://developer.nvidia.com/designworks/optix/download) (free
90
+ NVIDIA developer account required) and point `OPTIX_PATH` at its `include`
91
+ directory:
92
+
93
+ ```bash
94
+ OPTIX_PATH=/opt/NVIDIA-OptiX-SDK-8.0.0/include python3 pantheon.py --test rt_virus --duration 60
95
+ ```
96
+
97
+ The build cache keys on `OPTIX_PATH`, so setting or changing it triggers a
98
+ rebuild automatically. On AMD hardware, build with `ENABLE_HIPRT=1` instead,
99
+ which clones and builds HIP-RT itself.
100
+
101
+ ## AI and inference workloads
102
+
103
+ The `inference`, `training` and `ai_auxiliary` suites exercise GPU execution
104
+ and memory patterns relevant to model serving. They are diagnostic workloads,
105
+ not model benchmarks: they load no weights, measure no model quality, and do
106
+ not replace an end-to-end benchmark such as vLLM or TensorRT-LLM.
107
+
108
+ ```bash
109
+ python3 pantheon.py --test inference --duration 60 --gpu 0 --mem 50
110
+ ```
111
+
112
+ See [`docs/ai_workloads.md`](docs/ai_workloads.md) for what each workload
113
+ stresses and what its reported rate does and does not mean.
114
+
115
+ ## Hardware Counter Profiling
116
+
117
+ Add `--profile` to wrap each workload in the native vendor profiler:
118
+
119
+ ```bash
120
+ python3 pantheon.py --test tensor_virus --duration 30 --gpu 0 --profile
121
+ ```
122
+
123
+ `--profile` is an exhaustive diagnostic mode. Pantheon profiles every selected
124
+ workload and GPU, running GPUs serially to avoid profiler contention and to
125
+ keep the artifacts isolated. On NVIDIA it runs Nsight Compute (`ncu`) for
126
+ hardware counters and Nsight Systems (`nsys`) for a CUDA/NVTX timeline. On AMD
127
+ it uses `rocprofv3` to collect performance counters together with the HIP
128
+ runtime, kernel-dispatch, and memory-operation trace.
129
+
130
+ The default counter sets cover SM/CU occupancy and active blocks, instruction
131
+ pipeline and memory-operation mix, branch and memory-access efficiency,
132
+ scheduler stalls, L1/L2/cache behavior, DRAM traffic, and host/peer transfer
133
+ fabric indicators where the installed profiler exposes them. Each profile also
134
+ includes the RAS and PCIe reliability snapshots collected before and after the
135
+ workload.
136
+ Supported devices report ECC correctable and uncorrectable counts, memory
137
+ retirement state, and PCIe replay, CRC, TLP/DLLP, and fatal/non-fatal counters.
138
+ AMD profiles also collect exposed per-block RAS data and CPER firmware records.
139
+ Linux PCIe AER counters and new AER/poison events are included when the host
140
+ exposes them. Unsupported hardware is labeled unavailable rather than reported
141
+ as zero. One workload/GPU bundle is written below the timestamped run directory:
142
+
143
+ ```text
144
+ results/<run-id>/profiles/<workload>/gpu<id>/
145
+ profile_manifest.json
146
+ time_series.csv
147
+ hardware_counters.csv
148
+ ras.json
149
+ summary.json
150
+ trace.*
151
+ ```
152
+
153
+ `summary.xlsx` links each result to its bundle and includes the flattened
154
+ counter values. Profiling can require multiple vendor passes and is therefore
155
+ substantially slower than a normal stress run. Pantheon refuses an explicit
156
+ profile request when the required vendor tools are unavailable, instead of
157
+ silently running without profiler artifacts.
158
+
159
+ ## Fault Maps
160
+
161
+ Console fault reporting is capped at a few lines per run so a widescale
162
+ corruption cannot flood the log. That keeps the console readable but means a
163
+ run reports an accurate error *count* and only a handful of error *locations*.
164
+
165
+ Memory workloads accept `--fault_map <file>` to record every failing address:
166
+
167
+ ```bash
168
+ ./build/memory_read 0 60 90 --verify --fault_map faults.csv
169
+ ```
170
+
171
+ The file is self-describing CSV:
172
+
173
+ ```text
174
+ # workload=memory_read gpu=0 init_pattern=2 init_pattern_name=crosstalk total_faults=3 recorded=3 capacity=4096
175
+ index,expected,actual,xor
176
+ 1337,0xe76e5272,0xecc3ec9d,0x0badbeef
177
+ ```
178
+
179
+ The `xor` column names the flipped bits, which is what distinguishes a single
180
+ defective cell from a stuck data lane: one bit position failing across many
181
+ unrelated addresses indicates shared structure, not a cell. Supported by
182
+ `memory_read`, `memory_write`, `atomic_virus`, `ras_validator`,
183
+ `memory_retention`, `march_test`, `memory_hammer`, and `galpat`
184
+ (`memory_read_agg` and `memory_write_agg` are aliases of the first two with a
185
+ different `--init_pattern`).
186
+
187
+ ## Memory Diagnostics
188
+
189
+ The `diagnostics` suite looks for defective cells rather than measuring
190
+ bandwidth: `march_test`, `memory_hammer`, `galpat`, `memory_retention` and
191
+ `ras_validator`.
192
+
193
+ ```bash
194
+ python3 pantheon.py --suite diagnostics --duration 300 --gpu 0 --mem 90
195
+ ```
196
+
197
+ See [`docs/memory_diagnostics.md`](docs/memory_diagnostics.md) for what each
198
+ one finds, how they compose as a funnel, the full `--init_pattern` table, and
199
+ the caveats that decide whether a result means anything.
200
+
201
+ ## Reliability checks on every run
202
+
203
+ Pantheon collects lightweight before-and-after reliability snapshots for every
204
+ workload, even without `--profile`. Each result records a `RAS Status`, a
205
+ human-readable `RAS Error Delta`, and the path to its JSON report under
206
+ `results/<run-id>/reliability/<workload>/gpu<id>/ras.json`.
207
+
208
+ The snapshot covers NVIDIA ECC and PCIe error counters, AMD RAS and CPER
209
+ records where their management tools expose them, and Linux PCIe AER events.
210
+ `CLEAN` means supported counters did not change during the workload;
211
+ `WARNING` records new correctable or recoverable events; `ERROR` records new
212
+ uncorrectable, fatal, or poison events; and `UNAVAILABLE` means the platform
213
+ does not expose the relevant telemetry. Hardware counters and traces still
214
+ require `--profile`.
215
+
216
+ Profiler counter names can vary by GPU generation and driver/toolkit version.
217
+ Override or extend the default sets without editing the source:
218
+
219
+ ```bash
220
+ # Replace the NVIDIA metric list entirely
221
+ PANTHEON_CUDA_METRICS="sm__throughput.avg.pct_of_peak_sustained_elapsed,dram__bytes_read.sum" \
222
+ python3 pantheon.py --test memory_read --duration 30 --profile
223
+
224
+ # Add one AMD counter to Pantheon's default list
225
+ PANTHEON_HIP_METRICS_APPEND="CUSTOM_COUNTER_NAME" \
226
+ python3 pantheon.py --test memory_write --duration 30 --profile
227
+ ```
228
+
229
+ ## Interpretation of Results
230
+
231
+ The summary report (`results/<timestamp>/summary.xlsx`) contains detailed "Pro" metrics. Here is how to interpret them:
232
+
233
+ * **Efficiency (MB/J):** Calculated as `Throughput / Watts`.
234
+ * **Healthy:** Stays relatively constant throughout the run.
235
+ * **Degraded:** If this drops significantly during a 1-hour burn-in, your silicon is "leaking" current (thermal runaway) or the VRMs are becoming inefficient due to heat.
236
+ * **PCIe Link:** Verifies the physical connection speed (e.g., `Gen4 x16`).
237
+ * **Red Flag:** If it drops to `x8` or `Gen3` under load, check your riser cable, motherboard slot, or GPU mounting pressure.
238
+ * **Throttle Reason:** Tells you *why* performance is limited.
239
+ * `[POWER]`: **Normal.** The card hit its TDP limit (expected for viruses).
240
+ * `[THERMAL]`: **Critical.** The core is overheating (typically >83°C Edge or >110°C Junction). Check thermal paste.
241
+ * `[VOLTAGE]`: The VRMs cannot supply enough stable voltage to maintain the clock.
242
+ * **Junction (Hotspot) vs. Edge:** * On newer AMD drivers, the primary temperature reported is the **Junction**. This is the absolute hottest single point on the silicon die.
243
+ * It is normal for Junction to be 15-25°C hotter than the traditional "Edge" average.
244
+ * **Max Mem Temp:** The hottest point on your VRAM (Memory/GDDR6X). Keep this under 100°C to avoid permanent damage.
245
+ * **Throughput (GB/s):**
246
+ * For `memory_read` / `memory_write`, this should be within 90% of your card's theoretical max bandwidth.
247
+ * Low throughput = Memory Controller instability or aggressive error correction (ECC) kicking in.
248
+
249
+ ## Incremental Reports
250
+
251
+ For long `--test all` runs, Pantheon writes one atomic
252
+ `database/pantheon_report_<run>_<sequence>_<workload>_gpu<id>.json` file when
253
+ each workload finishes. These are complete workload records, not snapshots of
254
+ an active workload, so a results publisher can safely copy them while the
255
+ remaining queue is still running. The aggregate report is still written at the
256
+ end of the run. Reports marked `partial`, `failed`, or `incomplete` are
257
+ excluded by downstream importers.
258
+
259
+ ## Per-workload documentation
260
+
261
+ Each workload has its own page under `kernels/<test>/README.md` describing what
262
+ it stresses, how to read a failure, and what its reported metric does not mean.
263
+
264
+ ## Disclaimer
265
+
266
+ Pantheon is an independent project. It is not generated, verified, or endorsed
267
+ by **AMD** or **NVIDIA**, and its results are not produced by official vendor
268
+ performance labs. Measurements reflect the driver versions, power limits and
269
+ thermal conditions of whatever machine they were taken on, so figures from
270
+ different environments are not directly comparable.