File size: 2,930 Bytes
18f6a27 | 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 | {
"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"
],
"gbr_importance": [
0.00020907692527896766,
1.8562661189242273e-07,
2.9369056959320215e-05,
0.0,
0.0,
0.0,
1.34552169063225e-06,
0.0,
1.2959966275380288e-08,
2.8763114459042767e-08,
9.104787556291898e-06,
5.54221245471016e-06,
4.32329418665271e-06,
0.0,
1.2805806474662726e-06,
2.257546676504885e-08,
1.735686571541027e-07,
8.901842573818885e-07,
2.311281409817864e-06,
9.20361423082855e-07,
2.7881975535483807e-06,
2.0856182994836325e-08,
1.1771724567239878e-06,
2.7156362706913446e-05,
8.131880276961555e-05,
7.180057550667766e-05,
0.0004996830995928254,
1.5779269357750438e-06,
0.0001825415270437779,
1.6970423039960086e-05,
0.00041935952304566224,
3.4117339243641743e-06,
6.432292741146238e-06,
0.0,
5.033034070243649e-06,
1.945656188484425e-06,
4.355968723707434e-05,
2.3658129881027048e-05,
7.174538256656928e-06,
5.067777589608545e-05,
4.413329058621823e-05,
6.258530983657731e-06,
0.0003196131073257341,
0.0009352059938410164,
0.005090926310276446,
0.0,
0.01927022907891746,
0.02657711487616414,
0.01086756094557264,
0.003634432430735974,
0.0007652333385088071,
0.007919310415035623,
0.001091000073631452,
0.012641277527748929,
0.010506180230114641,
0.010371283017597741,
0.015597399741008567,
0.002644217662689972,
0.008205515687632867,
0.0018993846795010236,
0.6869183995385755,
0.1375995391029925,
0.009656091846671428,
0.0257288175591774
]
} |