{ "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 ] }