File size: 5,353 Bytes
4803fb6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
{
  "mean": [
    29.028148192477364,
    0.7685549132947976,
    0.15260115606936417,
    0.006936416184971098,
    0.014797687861271676,
    0.0,
    0.34959537572254334,
    0.0,
    0.06936416184971098,
    0.004624277456647399,
    0.058265895953757224,
    0.09017341040462427,
    0.18867052023121386,
    0.006936416184971098,
    0.23028901734104046,
    0.04439306358381503,
    0.014797687861271676,
    0.0027745664739884392,
    0.09710982658959537,
    0.0027745664739884392,
    0.11560693641618497,
    0.0011560693641618498,
    0.0027745664739884392,
    0.09710982658959537,
    5061.267514450867,
    419113.38173410407,
    415367.2988493303,
    0.1692485549132948,
    413670.53156069363,
    0.2575722543352601,
    413.2030057803468,
    0.055491329479768786,
    4.4763005780346825,
    0.0,
    1.0554913294797688,
    1.8240462427745665,
    6.750520231213873,
    13.416647398843931,
    1.218034682080925,
    12.705202312138729,
    62.67052023121387,
    4.789364161849711,
    28.624801572237402,
    8.499549905082393,
    11.610112068970079,
    1.0,
    0.40277456647398846,
    0.5278612716763006,
    10453.333333333334,
    422.6666666666667,
    43.333333333333336,
    1223.25,
    1113.75,
    1810.0,
    96.33333333333333,
    40.649999698003136,
    269.94166628519696,
    324.3333333333333,
    8.416666666666666,
    35.333333333333336,
    18.910816084289582,
    16.392050143643036,
    243.22255797046367,
    3.1944508670520233
  ],
  "scale": [
    3.1041677066766096,
    0.4217561600560334,
    0.35960261850499764,
    0.0829957969747866,
    0.12074235501772981,
    1.0,
    0.4768421636092736,
    1.0,
    0.254072381223593,
    0.06784462775085053,
    0.23424555774329966,
    0.2864300376368528,
    0.3912466677517355,
    0.0829957969747866,
    0.4210177975256749,
    0.20596679220071845,
    0.12074235501772909,
    0.05260102902863372,
    0.2961072578803945,
    0.0526010290286337,
    0.3197529869580254,
    0.033981360593523975,
    0.05260102902863369,
    0.2961072578803945,
    11929.61616276313,
    6100473.100893826,
    6100716.323838187,
    0.37497130766634457,
    6100830.297789359,
    0.5497376165877587,
    197.26431068090426,
    0.22893676382863742,
    1.9545601524187768,
    1.0,
    0.22893676382863742,
    0.38078081955825765,
    2.747308144620515,
    5.0377350949857505,
    1.6858056371346821,
    5.0437423702852335,
    29.313894164003376,
    1.8667216945016867,
    8.382944881775504,
    1.9106927690961886,
    5.750476170008533,
    1.0,
    0.4904561296138203,
    0.4992231460384118,
    4831.140468070922,
    119.10033118717543,
    23.19961685506729,
    846.0870842689106,
    419.8269732878058,
    362.09460089871544,
    29.550333707463455,
    25.668446445301374,
    285.8037277789368,
    165.90124237697464,
    0.6242773141816064,
    28.986586936415584,
    4.992034798897117,
    2.4286824361046477,
    590.0480231826563,
    0.9809122592276406
  ],
  "feature_names": [
    "avg_line_length",
    "compute_bound_score",
    "has_attention",
    "has_backward",
    "has_batchnorm",
    "has_cat",
    "has_conv",
    "has_dropout",
    "has_embedding",
    "has_fft",
    "has_gelu",
    "has_layernorm",
    "has_linear",
    "has_loss",
    "has_matmul",
    "has_pooling",
    "has_reduce",
    "has_relu",
    "has_reshape",
    "has_sigmoid",
    "has_softmax",
    "has_sort",
    "has_tanh",
    "has_transpose",
    "max_dim",
    "max_numeric",
    "mean_numeric",
    "memory_bound_score",
    "min_numeric",
    "mixed_score",
    "num_chars",
    "num_class_defs",
    "num_dim_specs",
    "num_for_loops",
    "num_function_defs",
    "num_imports",
    "num_large_dims",
    "num_lines",
    "num_nn_calls",
    "num_numeric_literals",
    "num_tokens",
    "num_torch_calls",
    "product_large_dims_log",
    "sum_numeric_log",
    "total_elements_log",
    "uses_cuda",
    "uses_float16",
    "uses_float32",
    "cuda_cores",
    "tensor_cores",
    "memory_gb",
    "memory_bandwidth_gbps",
    "base_clock_mhz",
    "boost_clock_mhz",
    "sm_count",
    "fp32_tflops",
    "fp16_tflops",
    "tdp_watts",
    "compute_capability",
    "l2_cache_mb",
    "flops_log",
    "memory_bytes_log",
    "arithmetic_intensity",
    "dtype_bytes"
  ],
  "n_code_features": 48,
  "n_gpu_features": 12,
  "n_extra_features": 4,
  "code_feature_names": [
    "avg_line_length",
    "compute_bound_score",
    "has_attention",
    "has_backward",
    "has_batchnorm",
    "has_cat",
    "has_conv",
    "has_dropout",
    "has_embedding",
    "has_fft",
    "has_gelu",
    "has_layernorm",
    "has_linear",
    "has_loss",
    "has_matmul",
    "has_pooling",
    "has_reduce",
    "has_relu",
    "has_reshape",
    "has_sigmoid",
    "has_softmax",
    "has_sort",
    "has_tanh",
    "has_transpose",
    "max_dim",
    "max_numeric",
    "mean_numeric",
    "memory_bound_score",
    "min_numeric",
    "mixed_score",
    "num_chars",
    "num_class_defs",
    "num_dim_specs",
    "num_for_loops",
    "num_function_defs",
    "num_imports",
    "num_large_dims",
    "num_lines",
    "num_nn_calls",
    "num_numeric_literals",
    "num_tokens",
    "num_torch_calls",
    "product_large_dims_log",
    "sum_numeric_log",
    "total_elements_log",
    "uses_cuda",
    "uses_float16",
    "uses_float32"
  ]
}