GPUMODE/backendbench_tests
TorchBench The TorchBench suite of BackendBench is designed to mimic real-world use cases. It provides operators and inputs derived from 155 model traces found in TIMM (67), Hugging Face Transformers (45), and TorchBench (43). (These are also the models PyTorch developers use to validate performance.) You can view the origin of these traces by switching the subset in the dataset viewer to ops_traces_models and torchbench for the full dataset. When running BackendBench, much of… See the full description on the dataset page: https://huggingface.co/datasets/GPUMODE/backendbench_tests.
4805
1{2 "aten._adaptive_avg_pool2d.default": {3 "((T([128, 256, 6, 6], f16), [6, 6]), {})": [4 "TorchBench/alexnet"5 ],6 "((T([64, 512, 7, 7], f16), [7, 7]), {})": [7 "TorchBench/vgg16"8 ]9 },10 "aten._adaptive_avg_pool2d_backward.default": {11 "((T([128, 256, 6, 6], f16), T([128, 256, 6, 6], f16)), {})": [12 "TorchBench/alexnet"13 ],14 "((T([64, 512, 7, 7], f16), T([64, 512, 7, 7], f16)), {})": [15 "TorchBench/vgg16"16 ]17 },18 "aten._cudnn_rnn.default": {19 "((T([64, 50, 256], f16), [T([3072, 256], f16), T([3072, 768], f16), T([3072], f16), T([3072], f16)], 4, None, T([1, 64, 768], f16), T([1, 64, 768], f16), 2, 768, 0, 1, True, 0.0, True, False, [], None), {})": [20 "TorchBench/tts_angular"21 ],22 "((T([64, 50, 40], f16), [T([3072, 40], f16), T([3072, 768], f16), T([3072], f16), T([3072], f16)], 4, None, T([1, 64, 768], f16), T([1, 64, 768], f16), 2, 768, 0, 1, True, 0.0, True, False, [], None), {})": [23 "TorchBench/tts_angular"24 ]25 },26 "aten._cudnn_rnn_backward.default": {27 "((T([64, 50, 256], f16), [T([3072, 256], f16), T([3072, 768], f16), T([3072], f16), T([3072], f16)], 4, T([3151872], f16), T([1, 64, 768], f16), T([1, 64, 768], f16), T([64, 50, 768], f16, stride=(768, 49152, 1)), T([64, 50, 768], f16), None, None, 2, 768, 0, 1, True, 0.0, True, False, [], None, T([24576016], u8), [True, False, False, True]), {})": [28 "TorchBench/tts_angular"29 ],30 "((T([64, 50, 40], f16), [T([3072, 40], f16), T([3072, 768], f16), T([3072], f16), T([3072], f16)], 4, T([2488320], f16), T([1, 64, 768], f16), T([1, 64, 768], f16), T([64, 50, 768], f16, stride=(768, 49152, 1)), T([64, 50, 768], f16), None, None, 2, 768, 0, 1, True, 0.0, True, False, [], None, T([24576016], u8), [False, False, False, True]), {})": [31 "TorchBench/tts_angular"32 ]33 },34 "aten._embedding_bag.default": {35 "((T([965, 192], f16), T([54687], i64), T([1024], i64), False, 0, True, T([54687], f16)), {})": [36 "TorchBench/fambench_dlrm"37 ],38 "((T([965, 192], f16), T([54692], i64), T([1024], i64), False, 0, True, T([54692], f16)), {})": [39 "TorchBench/fambench_dlrm"40 ],41 "((T([965, 192], f16), T([54697], i64), T([1024], i64), False, 0, True, T([54697], f16)), {})": [42 "TorchBench/fambench_dlrm"43 ],44 "((T([965, 192], f16), T([54701], i64), T([1024], i64), False, 0, True, T([54701], f16)), {})": [45 "TorchBench/fambench_dlrm"46 ],47 "((T([965, 192], f16), T([54704], i64), T([1024], i64), False, 0, True, T([54704], f16)), {})": [48 "TorchBench/fambench_dlrm"49 ],50 "((T([965, 192], f16), T([54705], i64), T([1024], i64), False, 0, True, T([54705], f16)), {})": [51 "TorchBench/fambench_dlrm"52 ],53 "((T([965, 192], f16), T([54707], i64), T([1024], i64), False, 0, True, T([54707], f16)), {})": [54 "TorchBench/fambench_dlrm"55 ],56 "((T([965, 192], f16), T([54710], i64), T([1024], i64), False, 0, True, T([54710], f16)), {})": [57 "TorchBench/fambench_dlrm"58 ],59 "((T([965, 192], f16), T([54711], i64), T([1024], i64), False, 0, True, T([54711], f16)), {})": [60 "TorchBench/fambench_dlrm"61 ],62 "((T([965, 192], f16), T([54712], i64), T([1024], i64), False, 0, True, T([54712], f16)), {})": [63 "TorchBench/fambench_dlrm"64 ],65 "((T([965, 192], f16), T([54713], i64), T([1024], i64), False, 0, True, T([54713], f16)), {})": [66 "TorchBench/fambench_dlrm"67 ],68 "((T([965, 192], f16), T([54714], i64), T([1024], i64), False, 0, True, T([54714], f16)), {})": [69 "TorchBench/fambench_dlrm"70 ],71 "((T([965, 192], f16), T([54715], i64), T([1024], i64), False, 0, True, T([54715], f16)), {})": [72 "TorchBench/fambench_dlrm"73 ],74 "((T([965, 192], f16), T([54716], i64), T([1024], i64), False, 0, True, T([54716], f16)), {})": [75 "TorchBench/fambench_dlrm"76 ],77 "((T([965, 192], f16), T([54717], i64), T([1024], i64), False, 0, True, T([54717], f16)), {})": [78 "TorchBench/fambench_dlrm"79 ],80 "((T([965, 192], f16), T([54718], i64), T([1024], i64), False, 0, True, T([54718], f16)), {})": [81 "TorchBench/fambench_dlrm"82 ],83 "((T([965, 192], f16), T([54719], i64), T([1024], i64), False, 0, True, T([54719], f16)), {})": [84 "TorchBench/fambench_dlrm"85 ],86 "((T([965, 192], f16), T([54722], i64), T([1024], i64), False, 0, True, T([54722], f16)), {})": [87 "TorchBench/fambench_dlrm"88 ],89 "((T([965, 192], f16), T([54723], i64), T([1024], i64), False, 0, True, T([54723], f16)), {})": [90 "TorchBench/fambench_dlrm"91 ],92 "((T([965, 192], f16), T([54725], i64), T([1024], i64), False, 0, True, T([54725], f16)), {})": [93 "TorchBench/fambench_dlrm"94 ],95 "((T([965, 192], f16), T([54727], i64), T([1024], i64), False, 0, True, T([54727], f16)), {})": [96 "TorchBench/fambench_dlrm"97 ],98 "((T([965, 192], f16), T([54729], i64), T([1024], i64), False, 0, True, T([54729], f16)), {})": [99 "TorchBench/fambench_dlrm"100 ],101 "((T([965, 192], f16), T([54730], i64), T([1024], i64), False, 0, True, T([54730], f16)), {})": [102 "TorchBench/fambench_dlrm"103 ],104 "((T([965, 192], f16), T([54731], i64), T([1024], i64), False, 0, True, T([54731], f16)), {})": [105 "TorchBench/fambench_dlrm"106 ],107 "((T([965, 192], f16), T([54732], i64), T([1024], i64), False, 0, True, T([54732], f16)), {})": [108 "TorchBench/fambench_dlrm"109 ],110 "((T([965, 192], f16), T([54733], i64), T([1024], i64), False, 0, True, T([54733], f16)), {})": [111 "TorchBench/fambench_dlrm"112 ],113 "((T([965, 192], f16), T([54734], i64), T([1024], i64), False, 0, True, T([54734], f16)), {})": [114 "TorchBench/fambench_dlrm"115 ],116 "((T([965, 192], f16), T([54735], i64), T([1024], i64), False, 0, True, T([54735], f16)), {})": [117 "TorchBench/fambench_dlrm"118 ],119 "((T([965, 192], f16), T([54736], i64), T([1024], i64), False, 0, True, T([54736], f16)), {})": [120 "TorchBench/fambench_dlrm"121 ],122 "((T([965, 192], f16), T([54737], i64), T([1024], i64), False, 0, True, T([54737], f16)), {})": [123 "TorchBench/fambench_dlrm"124 ],125 "((T([965, 192], f16), T([54738], i64), T([1024], i64), False, 0, True, T([54738], f16)), {})": [126 "TorchBench/fambench_dlrm"127 ],128 "((T([965, 192], f16), T([54739], i64), T([1024], i64), False, 0, True, T([54739], f16)), {})": [129 "TorchBench/fambench_dlrm"130 ],131 "((T([965, 192], f16), T([54740], i64), T([1024], i64), False, 0, True, T([54740], f16)), {})": [132 "TorchBench/fambench_dlrm"133 ],134 "((T([965, 192], f16), T([54741], i64), T([1024], i64), False, 0, True, T([54741], f16)), {})": [135 "TorchBench/fambench_dlrm"136 ],137 "((T([965, 192], f16), T([54742], i64), T([1024], i64), False, 0, True, T([54742], f16)), {})": [138 "TorchBench/fambench_dlrm"139 ],140 "((T([965, 192], f16), T([54743], i64), T([1024], i64), False, 0, True, T([54743], f16)), {})": [141 "TorchBench/fambench_dlrm"142 ],143 "((T([965, 192], f16), T([54744], i64), T([1024], i64), False, 0, True, T([54744], f16)), {})": [144 "TorchBench/fambench_dlrm"145 ],146 "((T([965, 192], f16), T([54745], i64), T([1024], i64), False, 0, True, T([54745], f16)), {})": [147 "TorchBench/fambench_dlrm"148 ],149 "((T([965, 192], f16), T([54746], i64), T([1024], i64), False, 0, True, T([54746], f16)), {})": [150 "TorchBench/fambench_dlrm"151 ],152 "((T([965, 192], f16), T([54747], i64), T([1024], i64), False, 0, True, T([54747], f16)), {})": [153 "TorchBench/fambench_dlrm"154 ],155 "((T([965, 192], f16), T([54748], i64), T([1024], i64), False, 0, True, T([54748], f16)), {})": [156 "TorchBench/fambench_dlrm"157 ],158 "((T([965, 192], f16), T([54749], i64), T([1024], i64), False, 0, True, T([54749], f16)), {})": [159 "TorchBench/fambench_dlrm"160 ],161 "((T([965, 192], f16), T([54750], i64), T([1024], i64), False, 0, True, T([54750], f16)), {})": [162 "TorchBench/fambench_dlrm"163 ],164 "((T([965, 192], f16), T([54751], i64), T([1024], i64), False, 0, True, T([54751], f16)), {})": [165 "TorchBench/fambench_dlrm"166 ],167 "((T([965, 192], f16), T([54753], i64), T([1024], i64), False, 0, True, T([54753], f16)), {})": [168 "TorchBench/fambench_dlrm"169 ],170 "((T([965, 192], f16), T([54754], i64), T([1024], i64), False, 0, True, T([54754], f16)), {})": [171 "TorchBench/fambench_dlrm"172 ],173 "((T([965, 192], f16), T([54756], i64), T([1024], i64), False, 0, True, T([54756], f16)), {})": [174 "TorchBench/fambench_dlrm"175 ],176 "((T([965, 192], f16), T([54757], i64), T([1024], i64), False, 0, True, T([54757], f16)), {})": [177 "TorchBench/fambench_dlrm"178 ],179 "((T([965, 192], f16), T([54758], i64), T([1024], i64), False, 0, True, T([54758], f16)), {})": [180 "TorchBench/fambench_dlrm"181 ],182 "((T([965, 192], f16), T([54759], i64), T([1024], i64), False, 0, True, T([54759], f16)), {})": [183 "TorchBench/fambench_dlrm"184 ],185 "((T([965, 192], f16), T([54760], i64), T([1024], i64), False, 0, True, T([54760], f16)), {})": [186 "TorchBench/fambench_dlrm"187 ],188 "((T([965, 192], f16), T([54761], i64), T([1024], i64), False, 0, True, T([54761], f16)), {})": [189 "TorchBench/fambench_dlrm"190 ],191 "((T([965, 192], f16), T([54762], i64), T([1024], i64), False, 0, True, T([54762], f16)), {})": [192 "TorchBench/fambench_dlrm"193 ],194 "((T([965, 192], f16), T([54763], i64), T([1024], i64), False, 0, True, T([54763], f16)), {})": [195 "TorchBench/fambench_dlrm"196 ],197 "((T([965, 192], f16), T([54764], i64), T([1024], i64), False, 0, True, T([54764], f16)), {})": [198 "TorchBench/fambench_dlrm"199 ],200 "((T([965, 192], f16), T([54765], i64), T([1024], i64), False, 0, True, T([54765], f16)), {})": [201 "TorchBench/fambench_dlrm"202 ],203 "((T([965, 192], f16), T([54766], i64), T([1024], i64), False, 0, True, T([54766], f16)), {})": [204 "TorchBench/fambench_dlrm"205 ],206 "((T([965, 192], f16), T([54767], i64), T([1024], i64), False, 0, True, T([54767], f16)), {})": [207 "TorchBench/fambench_dlrm"208 ],209 "((T([965, 192], f16), T([54768], i64), T([1024], i64), False, 0, True, T([54768], f16)), {})": [210 "TorchBench/fambench_dlrm"211 ],212 "((T([965, 192], f16), T([54769], i64), T([1024], i64), False, 0, True, T([54769], f16)), {})": [213 "TorchBench/fambench_dlrm"214 ],215 "((T([965, 192], f16), T([54770], i64), T([1024], i64), False, 0, True, T([54770], f16)), {})": [216 "TorchBench/fambench_dlrm"217 ],218 "((T([965, 192], f16), T([54771], i64), T([1024], i64), False, 0, True, T([54771], f16)), {})": [219 "TorchBench/fambench_dlrm"220 ],221 "((T([965, 192], f16), T([54772], i64), T([1024], i64), False, 0, True, T([54772], f16)), {})": [222 "TorchBench/fambench_dlrm"223 ],224 "((T([965, 192], f16), T([54773], i64), T([1024], i64), False, 0, True, T([54773], f16)), {})": [225 "TorchBench/fambench_dlrm"226 ],227 "((T([965, 192], f16), T([54774], i64), T([1024], i64), False, 0, True, T([54774], f16)), {})": [228 "TorchBench/fambench_dlrm"229 ],230 "((T([965, 192], f16), T([54775], i64), T([1024], i64), False, 0, True, T([54775], f16)), {})": [231 "TorchBench/fambench_dlrm"232 ],233 "((T([965, 192], f16), T([54776], i64), T([1024], i64), False, 0, True, T([54776], f16)), {})": [234 "TorchBench/fambench_dlrm"235 ],236 "((T([965, 192], f16), T([54777], i64), T([1024], i64), False, 0, True, T([54777], f16)), {})": [237 "TorchBench/fambench_dlrm"238 ],239 "((T([965, 192], f16), T([54778], i64), T([1024], i64), False, 0, True, T([54778], f16)), {})": [240 "TorchBench/fambench_dlrm"241 ],242 "((T([965, 192], f16), T([54779], i64), T([1024], i64), False, 0, True, T([54779], f16)), {})": [243 "TorchBench/fambench_dlrm"244 ],245 "((T([965, 192], f16), T([54780], i64), T([1024], i64), False, 0, True, T([54780], f16)), {})": [246 "TorchBench/fambench_dlrm"247 ],248 "((T([965, 192], f16), T([54781], i64), T([1024], i64), False, 0, True, T([54781], f16)), {})": [249 "TorchBench/fambench_dlrm"250 ],251 "((T([965, 192], f16), T([54782], i64), T([1024], i64), False, 0, True, T([54782], f16)), {})": [252 "TorchBench/fambench_dlrm"253 ],254 "((T([965, 192], f16), T([54783], i64), T([1024], i64), False, 0, True, T([54783], f16)), {})": [255 "TorchBench/fambench_dlrm"256 ],257 "((T([965, 192], f16), T([54784], i64), T([1024], i64), False, 0, True, T([54784], f16)), {})": [258 "TorchBench/fambench_dlrm"259 ],260 "((T([965, 192], f16), T([54785], i64), T([1024], i64), False, 0, True, T([54785], f16)), {})": [261 "TorchBench/fambench_dlrm"262 ],263 "((T([965, 192], f16), T([54786], i64), T([1024], i64), False, 0, True, T([54786], f16)), {})": [264 "TorchBench/fambench_dlrm"265 ],266 "((T([965, 192], f16), T([54787], i64), T([1024], i64), False, 0, True, T([54787], f16)), {})": [267 "TorchBench/fambench_dlrm"268 ],269 "((T([965, 192], f16), T([54788], i64), T([1024], i64), False, 0, True, T([54788], f16)), {})": [270 "TorchBench/fambench_dlrm"271 ],272 "((T([965, 192], f16), T([54789], i64), T([1024], i64), False, 0, True, T([54789], f16)), {})": [273 "TorchBench/fambench_dlrm"274 ],275 "((T([965, 192], f16), T([54791], i64), T([1024], i64), False, 0, True, T([54791], f16)), {})": [276 "TorchBench/fambench_dlrm"277 ],278 "((T([965, 192], f16), T([54792], i64), T([1024], i64), False, 0, True, T([54792], f16)), {})": [279 "TorchBench/fambench_dlrm"280 ],281 "((T([965, 192], f16), T([54793], i64), T([1024], i64), False, 0, True, T([54793], f16)), {})": [282 "TorchBench/fambench_dlrm"283 ],284 "((T([965, 192], f16), T([54794], i64), T([1024], i64), False, 0, True, T([54794], f16)), {})": [285 "TorchBench/fambench_dlrm"286 ],287 "((T([965, 192], f16), T([54795], i64), T([1024], i64), False, 0, True, T([54795], f16)), {})": [288 "TorchBench/fambench_dlrm"289 ],290 "((T([965, 192], f16), T([54796], i64), T([1024], i64), False, 0, True, T([54796], f16)), {})": [291 "TorchBench/fambench_dlrm"292 ],293 "((T([965, 192], f16), T([54797], i64), T([1024], i64), False, 0, True, T([54797], f16)), {})": [294 "TorchBench/fambench_dlrm"295 ],296 "((T([965, 192], f16), T([54798], i64), T([1024], i64), False, 0, True, T([54798], f16)), {})": [297 "TorchBench/fambench_dlrm"298 ],299 "((T([965, 192], f16), T([54799], i64), T([1024], i64), False, 0, True, T([54799], f16)), {})": [300 "TorchBench/fambench_dlrm"301 ],302 "((T([965, 192], f16), T([54800], i64), T([1024], i64), False, 0, True, T([54800], f16)), {})": [303 "TorchBench/fambench_dlrm"304 ],305 "((T([965, 192], f16), T([54801], i64), T([1024], i64), False, 0, True, T([54801], f16)), {})": [306 "TorchBench/fambench_dlrm"307 ],308 "((T([965, 192], f16), T([54802], i64), T([1024], i64), False, 0, True, T([54802], f16)), {})": [309 "TorchBench/fambench_dlrm"310 ],311 "((T([965, 192], f16), T([54803], i64), T([1024], i64), False, 0, True, T([54803], f16)), {})": [312 "TorchBench/fambench_dlrm"313 ],314 "((T([965, 192], f16), T([54804], i64), T([1024], i64), False, 0, True, T([54804], f16)), {})": [315 "TorchBench/fambench_dlrm"316 ],317 "((T([965, 192], f16), T([54807], i64), T([1024], i64), False, 0, True, T([54807], f16)), {})": [318 "TorchBench/fambench_dlrm"319 ],320 "((T([965, 192], f16), T([54809], i64), T([1024], i64), False, 0, True, T([54809], f16)), {})": [321 "TorchBench/fambench_dlrm"322 ],323 "((T([965, 192], f16), T([54811], i64), T([1024], i64), False, 0, True, T([54811], f16)), {})": [324 "TorchBench/fambench_dlrm"325 ],326 "((T([965, 192], f16), T([54812], i64), T([1024], i64), False, 0, True, T([54812], f16)), {})": [327 "TorchBench/fambench_dlrm"328 ],329 "((T([965, 192], f16), T([54813], i64), T([1024], i64), False, 0, True, T([54813], f16)), {})": [330 "TorchBench/fambench_dlrm"331 ],332 "((T([965, 192], f16), T([54814], i64), T([1024], i64), False, 0, True, T([54814], f16)), {})": [333 "TorchBench/fambench_dlrm"334 ],335 "((T([965, 192], f16), T([54815], i64), T([1024], i64), False, 0, True, T([54815], f16)), {})": [336 "TorchBench/fambench_dlrm"337 ],338 "((T([965, 192], f16), T([54816], i64), T([1024], i64), False, 0, True, T([54816], f16)), {})": [339 "TorchBench/fambench_dlrm"340 ],341 "((T([965, 192], f16), T([54818], i64), T([1024], i64), False, 0, True, T([54818], f16)), {})": [342 "TorchBench/fambench_dlrm"343 ],344 "((T([965, 192], f16), T([54819], i64), T([1024], i64), False, 0, True, T([54819], f16)), {})": [345 "TorchBench/fambench_dlrm"346 ],347 "((T([965, 192], f16), T([54820], i64), T([1024], i64), False, 0, True, T([54820], f16)), {})": [348 "TorchBench/fambench_dlrm"349 ],350 "((T([965, 192], f16), T([54821], i64), T([1024], i64), False, 0, True, T([54821], f16)), {})": [351 "TorchBench/fambench_dlrm"352 ],353 "((T([965, 192], f16), T([54824], i64), T([1024], i64), False, 0, True, T([54824], f16)), {})": [354 "TorchBench/fambench_dlrm"355 ],356 "((T([965, 192], f16), T([54826], i64), T([1024], i64), False, 0, True, T([54826], f16)), {})": [357 "TorchBench/fambench_dlrm"358 ],359 "((T([965, 192], f16), T([54827], i64), T([1024], i64), False, 0, True, T([54827], f16)), {})": [360 "TorchBench/fambench_dlrm"361 ],362 "((T([965, 192], f16), T([54828], i64), T([1024], i64), False, 0, True, T([54828], f16)), {})": [363 "TorchBench/fambench_dlrm"364 ],365 "((T([965, 192], f16), T([54829], i64), T([1024], i64), False, 0, True, T([54829], f16)), {})": [366 "TorchBench/fambench_dlrm"367 ],368 "((T([965, 192], f16), T([54830], i64), T([1024], i64), False, 0, True, T([54830], f16)), {})": [369 "TorchBench/fambench_dlrm"370 ],371 "((T([965, 192], f16), T([54831], i64), T([1024], i64), False, 0, True, T([54831], f16)), {})": [372 "TorchBench/fambench_dlrm"373 ],374 "((T([965, 192], f16), T([54833], i64), T([1024], i64), False, 0, True, T([54833], f16)), {})": [375 "TorchBench/fambench_dlrm"376 ],377 "((T([965, 192], f16), T([54835], i64), T([1024], i64), False, 0, True, T([54835], f16)), {})": [378 "TorchBench/fambench_dlrm"379 ],380 "((T([965, 192], f16), T([54836], i64), T([1024], i64), False, 0, True, T([54836], f16)), {})": [381 "TorchBench/fambench_dlrm"382 ],383 "((T([965, 192], f16), T([54838], i64), T([1024], i64), False, 0, True, T([54838], f16)), {})": [384 "TorchBench/fambench_dlrm"385 ],386 "((T([965, 192], f16), T([54839], i64), T([1024], i64), False, 0, True, T([54839], f16)), {})": [387 "TorchBench/fambench_dlrm"388 ],389 "((T([965, 192], f16), T([54842], i64), T([1024], i64), False, 0, True, T([54842], f16)), {})": [390 "TorchBench/fambench_dlrm"391 ],392 "((T([965, 192], f16), T([54843], i64), T([1024], i64), False, 0, True, T([54843], f16)), {})": [393 "TorchBench/fambench_dlrm"394 ],395 "((T([965, 192], f16), T([54844], i64), T([1024], i64), False, 0, True, T([54844], f16)), {})": [396 "TorchBench/fambench_dlrm"397 ],398 "((T([965, 192], f16), T([54847], i64), T([1024], i64), False, 0, True, T([54847], f16)), {})": [399 "TorchBench/fambench_dlrm"400 ],401 "((T([965, 192], f16), T([54855], i64), T([1024], i64), False, 0, True, T([54855], f16)), {})": [402 "TorchBench/fambench_dlrm"403 ],404 "((T([965, 192], f16), T([54862], i64), T([1024], i64), False, 0, True, T([54862], f16)), {})": [405 "TorchBench/fambench_dlrm"406 ]407 },408 "aten._embedding_bag_per_sample_weights_backward.default": {409 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54687], i64), T([1024], i64), T([54687], i64), 0), {})": [410 "TorchBench/fambench_dlrm"411 ],412 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54692], i64), T([1024], i64), T([54692], i64), 0), {})": [413 "TorchBench/fambench_dlrm"414 ],415 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54697], i64), T([1024], i64), T([54697], i64), 0), {})": [416 "TorchBench/fambench_dlrm"417 ],418 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54701], i64), T([1024], i64), T([54701], i64), 0), {})": [419 "TorchBench/fambench_dlrm"420 ],421 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54704], i64), T([1024], i64), T([54704], i64), 0), {})": [422 "TorchBench/fambench_dlrm"423 ],424 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54705], i64), T([1024], i64), T([54705], i64), 0), {})": [425 "TorchBench/fambench_dlrm"426 ],427 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54707], i64), T([1024], i64), T([54707], i64), 0), {})": [428 "TorchBench/fambench_dlrm"429 ],430 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54710], i64), T([1024], i64), T([54710], i64), 0), {})": [431 "TorchBench/fambench_dlrm"432 ],433 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54711], i64), T([1024], i64), T([54711], i64), 0), {})": [434 "TorchBench/fambench_dlrm"435 ],436 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54712], i64), T([1024], i64), T([54712], i64), 0), {})": [437 "TorchBench/fambench_dlrm"438 ],439 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54713], i64), T([1024], i64), T([54713], i64), 0), {})": [440 "TorchBench/fambench_dlrm"441 ],442 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54714], i64), T([1024], i64), T([54714], i64), 0), {})": [443 "TorchBench/fambench_dlrm"444 ],445 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54715], i64), T([1024], i64), T([54715], i64), 0), {})": [446 "TorchBench/fambench_dlrm"447 ],448 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54716], i64), T([1024], i64), T([54716], i64), 0), {})": [449 "TorchBench/fambench_dlrm"450 ],451 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54717], i64), T([1024], i64), T([54717], i64), 0), {})": [452 "TorchBench/fambench_dlrm"453 ],454 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54718], i64), T([1024], i64), T([54718], i64), 0), {})": [455 "TorchBench/fambench_dlrm"456 ],457 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54719], i64), T([1024], i64), T([54719], i64), 0), {})": [458 "TorchBench/fambench_dlrm"459 ],460 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54722], i64), T([1024], i64), T([54722], i64), 0), {})": [461 "TorchBench/fambench_dlrm"462 ],463 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54723], i64), T([1024], i64), T([54723], i64), 0), {})": [464 "TorchBench/fambench_dlrm"465 ],466 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54725], i64), T([1024], i64), T([54725], i64), 0), {})": [467 "TorchBench/fambench_dlrm"468 ],469 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54727], i64), T([1024], i64), T([54727], i64), 0), {})": [470 "TorchBench/fambench_dlrm"471 ],472 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54729], i64), T([1024], i64), T([54729], i64), 0), {})": [473 "TorchBench/fambench_dlrm"474 ],475 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54730], i64), T([1024], i64), T([54730], i64), 0), {})": [476 "TorchBench/fambench_dlrm"477 ],478 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54731], i64), T([1024], i64), T([54731], i64), 0), {})": [479 "TorchBench/fambench_dlrm"480 ],481 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54732], i64), T([1024], i64), T([54732], i64), 0), {})": [482 "TorchBench/fambench_dlrm"483 ],484 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54733], i64), T([1024], i64), T([54733], i64), 0), {})": [485 "TorchBench/fambench_dlrm"486 ],487 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54734], i64), T([1024], i64), T([54734], i64), 0), {})": [488 "TorchBench/fambench_dlrm"489 ],490 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54735], i64), T([1024], i64), T([54735], i64), 0), {})": [491 "TorchBench/fambench_dlrm"492 ],493 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54736], i64), T([1024], i64), T([54736], i64), 0), {})": [494 "TorchBench/fambench_dlrm"495 ],496 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54737], i64), T([1024], i64), T([54737], i64), 0), {})": [497 "TorchBench/fambench_dlrm"498 ],499 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54738], i64), T([1024], i64), T([54738], i64), 0), {})": [500 "TorchBench/fambench_dlrm"501 ],502 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54739], i64), T([1024], i64), T([54739], i64), 0), {})": [503 "TorchBench/fambench_dlrm"504 ],505 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54740], i64), T([1024], i64), T([54740], i64), 0), {})": [506 "TorchBench/fambench_dlrm"507 ],508 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54741], i64), T([1024], i64), T([54741], i64), 0), {})": [509 "TorchBench/fambench_dlrm"510 ],511 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54742], i64), T([1024], i64), T([54742], i64), 0), {})": [512 "TorchBench/fambench_dlrm"513 ],514 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54743], i64), T([1024], i64), T([54743], i64), 0), {})": [515 "TorchBench/fambench_dlrm"516 ],517 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54744], i64), T([1024], i64), T([54744], i64), 0), {})": [518 "TorchBench/fambench_dlrm"519 ],520 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54745], i64), T([1024], i64), T([54745], i64), 0), {})": [521 "TorchBench/fambench_dlrm"522 ],523 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54746], i64), T([1024], i64), T([54746], i64), 0), {})": [524 "TorchBench/fambench_dlrm"525 ],526 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54747], i64), T([1024], i64), T([54747], i64), 0), {})": [527 "TorchBench/fambench_dlrm"528 ],529 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54748], i64), T([1024], i64), T([54748], i64), 0), {})": [530 "TorchBench/fambench_dlrm"531 ],532 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54749], i64), T([1024], i64), T([54749], i64), 0), {})": [533 "TorchBench/fambench_dlrm"534 ],535 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54750], i64), T([1024], i64), T([54750], i64), 0), {})": [536 "TorchBench/fambench_dlrm"537 ],538 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54751], i64), T([1024], i64), T([54751], i64), 0), {})": [539 "TorchBench/fambench_dlrm"540 ],541 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54753], i64), T([1024], i64), T([54753], i64), 0), {})": [542 "TorchBench/fambench_dlrm"543 ],544 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54754], i64), T([1024], i64), T([54754], i64), 0), {})": [545 "TorchBench/fambench_dlrm"546 ],547 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54756], i64), T([1024], i64), T([54756], i64), 0), {})": [548 "TorchBench/fambench_dlrm"549 ],550 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54757], i64), T([1024], i64), T([54757], i64), 0), {})": [551 "TorchBench/fambench_dlrm"552 ],553 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54758], i64), T([1024], i64), T([54758], i64), 0), {})": [554 "TorchBench/fambench_dlrm"555 ],556 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54759], i64), T([1024], i64), T([54759], i64), 0), {})": [557 "TorchBench/fambench_dlrm"558 ],559 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54760], i64), T([1024], i64), T([54760], i64), 0), {})": [560 "TorchBench/fambench_dlrm"561 ],562 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54761], i64), T([1024], i64), T([54761], i64), 0), {})": [563 "TorchBench/fambench_dlrm"564 ],565 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54762], i64), T([1024], i64), T([54762], i64), 0), {})": [566 "TorchBench/fambench_dlrm"567 ],568 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54763], i64), T([1024], i64), T([54763], i64), 0), {})": [569 "TorchBench/fambench_dlrm"570 ],571 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54764], i64), T([1024], i64), T([54764], i64), 0), {})": [572 "TorchBench/fambench_dlrm"573 ],574 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54765], i64), T([1024], i64), T([54765], i64), 0), {})": [575 "TorchBench/fambench_dlrm"576 ],577 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54766], i64), T([1024], i64), T([54766], i64), 0), {})": [578 "TorchBench/fambench_dlrm"579 ],580 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54767], i64), T([1024], i64), T([54767], i64), 0), {})": [581 "TorchBench/fambench_dlrm"582 ],583 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54768], i64), T([1024], i64), T([54768], i64), 0), {})": [584 "TorchBench/fambench_dlrm"585 ],586 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54769], i64), T([1024], i64), T([54769], i64), 0), {})": [587 "TorchBench/fambench_dlrm"588 ],589 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54770], i64), T([1024], i64), T([54770], i64), 0), {})": [590 "TorchBench/fambench_dlrm"591 ],592 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54771], i64), T([1024], i64), T([54771], i64), 0), {})": [593 "TorchBench/fambench_dlrm"594 ],595 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54772], i64), T([1024], i64), T([54772], i64), 0), {})": [596 "TorchBench/fambench_dlrm"597 ],598 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54773], i64), T([1024], i64), T([54773], i64), 0), {})": [599 "TorchBench/fambench_dlrm"600 ],601 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54774], i64), T([1024], i64), T([54774], i64), 0), {})": [602 "TorchBench/fambench_dlrm"603 ],604 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54775], i64), T([1024], i64), T([54775], i64), 0), {})": [605 "TorchBench/fambench_dlrm"606 ],607 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54776], i64), T([1024], i64), T([54776], i64), 0), {})": [608 "TorchBench/fambench_dlrm"609 ],610 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54777], i64), T([1024], i64), T([54777], i64), 0), {})": [611 "TorchBench/fambench_dlrm"612 ],613 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54778], i64), T([1024], i64), T([54778], i64), 0), {})": [614 "TorchBench/fambench_dlrm"615 ],616 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54779], i64), T([1024], i64), T([54779], i64), 0), {})": [617 "TorchBench/fambench_dlrm"618 ],619 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54780], i64), T([1024], i64), T([54780], i64), 0), {})": [620 "TorchBench/fambench_dlrm"621 ],622 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54781], i64), T([1024], i64), T([54781], i64), 0), {})": [623 "TorchBench/fambench_dlrm"624 ],625 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54782], i64), T([1024], i64), T([54782], i64), 0), {})": [626 "TorchBench/fambench_dlrm"627 ],628 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54783], i64), T([1024], i64), T([54783], i64), 0), {})": [629 "TorchBench/fambench_dlrm"630 ],631 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54784], i64), T([1024], i64), T([54784], i64), 0), {})": [632 "TorchBench/fambench_dlrm"633 ],634 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54785], i64), T([1024], i64), T([54785], i64), 0), {})": [635 "TorchBench/fambench_dlrm"636 ],637 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54786], i64), T([1024], i64), T([54786], i64), 0), {})": [638 "TorchBench/fambench_dlrm"639 ],640 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54787], i64), T([1024], i64), T([54787], i64), 0), {})": [641 "TorchBench/fambench_dlrm"642 ],643 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54788], i64), T([1024], i64), T([54788], i64), 0), {})": [644 "TorchBench/fambench_dlrm"645 ],646 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54789], i64), T([1024], i64), T([54789], i64), 0), {})": [647 "TorchBench/fambench_dlrm"648 ],649 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54791], i64), T([1024], i64), T([54791], i64), 0), {})": [650 "TorchBench/fambench_dlrm"651 ],652 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54792], i64), T([1024], i64), T([54792], i64), 0), {})": [653 "TorchBench/fambench_dlrm"654 ],655 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54793], i64), T([1024], i64), T([54793], i64), 0), {})": [656 "TorchBench/fambench_dlrm"657 ],658 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54794], i64), T([1024], i64), T([54794], i64), 0), {})": [659 "TorchBench/fambench_dlrm"660 ],661 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54795], i64), T([1024], i64), T([54795], i64), 0), {})": [662 "TorchBench/fambench_dlrm"663 ],664 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54796], i64), T([1024], i64), T([54796], i64), 0), {})": [665 "TorchBench/fambench_dlrm"666 ],667 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54797], i64), T([1024], i64), T([54797], i64), 0), {})": [668 "TorchBench/fambench_dlrm"669 ],670 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54798], i64), T([1024], i64), T([54798], i64), 0), {})": [671 "TorchBench/fambench_dlrm"672 ],673 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54799], i64), T([1024], i64), T([54799], i64), 0), {})": [674 "TorchBench/fambench_dlrm"675 ],676 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54800], i64), T([1024], i64), T([54800], i64), 0), {})": [677 "TorchBench/fambench_dlrm"678 ],679 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54801], i64), T([1024], i64), T([54801], i64), 0), {})": [680 "TorchBench/fambench_dlrm"681 ],682 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54802], i64), T([1024], i64), T([54802], i64), 0), {})": [683 "TorchBench/fambench_dlrm"684 ],685 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54803], i64), T([1024], i64), T([54803], i64), 0), {})": [686 "TorchBench/fambench_dlrm"687 ],688 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54804], i64), T([1024], i64), T([54804], i64), 0), {})": [689 "TorchBench/fambench_dlrm"690 ],691 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54807], i64), T([1024], i64), T([54807], i64), 0), {})": [692 "TorchBench/fambench_dlrm"693 ],694 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54809], i64), T([1024], i64), T([54809], i64), 0), {})": [695 "TorchBench/fambench_dlrm"696 ],697 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54811], i64), T([1024], i64), T([54811], i64), 0), {})": [698 "TorchBench/fambench_dlrm"699 ],700 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54812], i64), T([1024], i64), T([54812], i64), 0), {})": [701 "TorchBench/fambench_dlrm"702 ],703 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54813], i64), T([1024], i64), T([54813], i64), 0), {})": [704 "TorchBench/fambench_dlrm"705 ],706 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54814], i64), T([1024], i64), T([54814], i64), 0), {})": [707 "TorchBench/fambench_dlrm"708 ],709 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54815], i64), T([1024], i64), T([54815], i64), 0), {})": [710 "TorchBench/fambench_dlrm"711 ],712 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54816], i64), T([1024], i64), T([54816], i64), 0), {})": [713 "TorchBench/fambench_dlrm"714 ],715 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54818], i64), T([1024], i64), T([54818], i64), 0), {})": [716 "TorchBench/fambench_dlrm"717 ],718 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54819], i64), T([1024], i64), T([54819], i64), 0), {})": [719 "TorchBench/fambench_dlrm"720 ],721 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54820], i64), T([1024], i64), T([54820], i64), 0), {})": [722 "TorchBench/fambench_dlrm"723 ],724 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54821], i64), T([1024], i64), T([54821], i64), 0), {})": [725 "TorchBench/fambench_dlrm"726 ],727 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54824], i64), T([1024], i64), T([54824], i64), 0), {})": [728 "TorchBench/fambench_dlrm"729 ],730 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54826], i64), T([1024], i64), T([54826], i64), 0), {})": [731 "TorchBench/fambench_dlrm"732 ],733 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54827], i64), T([1024], i64), T([54827], i64), 0), {})": [734 "TorchBench/fambench_dlrm"735 ],736 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54828], i64), T([1024], i64), T([54828], i64), 0), {})": [737 "TorchBench/fambench_dlrm"738 ],739 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54829], i64), T([1024], i64), T([54829], i64), 0), {})": [740 "TorchBench/fambench_dlrm"741 ],742 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54830], i64), T([1024], i64), T([54830], i64), 0), {})": [743 "TorchBench/fambench_dlrm"744 ],745 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54831], i64), T([1024], i64), T([54831], i64), 0), {})": [746 "TorchBench/fambench_dlrm"747 ],748 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54833], i64), T([1024], i64), T([54833], i64), 0), {})": [749 "TorchBench/fambench_dlrm"750 ],751 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54835], i64), T([1024], i64), T([54835], i64), 0), {})": [752 "TorchBench/fambench_dlrm"753 ],754 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54836], i64), T([1024], i64), T([54836], i64), 0), {})": [755 "TorchBench/fambench_dlrm"756 ],757 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54838], i64), T([1024], i64), T([54838], i64), 0), {})": [758 "TorchBench/fambench_dlrm"759 ],760 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54839], i64), T([1024], i64), T([54839], i64), 0), {})": [761 "TorchBench/fambench_dlrm"762 ],763 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54842], i64), T([1024], i64), T([54842], i64), 0), {})": [764 "TorchBench/fambench_dlrm"765 ],766 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54843], i64), T([1024], i64), T([54843], i64), 0), {})": [767 "TorchBench/fambench_dlrm"768 ],769 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54844], i64), T([1024], i64), T([54844], i64), 0), {})": [770 "TorchBench/fambench_dlrm"771 ],772 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54847], i64), T([1024], i64), T([54847], i64), 0), {})": [773 "TorchBench/fambench_dlrm"774 ],775 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54855], i64), T([1024], i64), T([54855], i64), 0), {})": [776 "TorchBench/fambench_dlrm"777 ],778 "((T([1024, 192], f16, stride=(47808, 1)), T([965, 192], f16), T([54862], i64), T([1024], i64), T([54862], i64), 0), {})": [779 "TorchBench/fambench_dlrm"780 ]781 },782 "aten._fft_c2c.default": {783 "((T([1, 512, 768], c32), [1, 2], 0, False), {})": [784 "HuggingFace/GoogleFnet"785 ],786 "((T([1, 512, 768], c32), [1, 2], 0, True), {})": [787 "HuggingFace/GoogleFnet"788 ]789 },790 "aten._index_put_impl_.default": {791 "((T([0, 4], f16), [T([0], i64)], T([0, 4], f16), True, True), {})": [792 "TorchBench/vision_maskrcnn"793 ],794 "((T([0], f16), [T([0], i64)], T([0], f16), True, True), {})": [795 "TorchBench/vision_maskrcnn"796 ],797 "((T([5000, 1], f32), [T([100], i64)], T([100, 1], f32, stride=(0, 0)), True, True), {})": [798 "TorchBench/timm_efficientdet"799 ],800 "((T([5000, 4], f32), [T([100], i64)], T([100, 4], f32), True, True), {})": [801 "TorchBench/timm_efficientdet"802 ],803 "((T([6, 474, 768], f16), [T([6, 474], i64, stride=(1, 0)), T([6, 474], i64, stride=(475, 1))], T([6, 474, 768], f16), True, True), {})": [804 "TorchBench/fastNLP_Bert"805 ]806 },807 "aten._log_softmax.default": {808 "((T([1, 2], f16), 1, False), {})": [809 "HuggingFace/GPTNeoForSequenceClassification"810 ],811 "((T([1, 512], f16), 1, False), {})": [812 "HuggingFace/DebertaV2ForQuestionAnswering"813 ],814 "((T([1024, 30000], f16), 1, False), {})": [815 "HuggingFace/AlbertForMaskedLM"816 ],817 "((T([1024, 50005], f16), 1, False), {})": [818 "HuggingFace/PLBartForConditionalGeneration"819 ],820 "((T([1024, 50265], f16), 1, False), {})": [821 "HuggingFace/AllenaiLongformerBase",822 "HuggingFace/MBartForConditionalGeneration",823 "HuggingFace/PegasusForCausalLM",824 "HuggingFace/TrOCRForCausalLM"825 ],826 "((T([1024, 50358], f16), 1, False), {})": [827 "HuggingFace/BigBird"828 ],829 "((T([127, 50257], f32), 1, False), {})": [830 "HuggingFace/GPTNeoForCausalLM"831 ],832 "((T([128, 1000], f16), 1, False), {})": [833 "TIMM/adv_inception_v3",834 "TIMM/botnet26t_256",835 "TIMM/coat_lite_mini",836 "TIMM/dm_nfnet_f0",837 "TIMM/eca_botnext26ts_256",838 "TIMM/eca_halonext26ts",839 "TIMM/ese_vovnet19b_dw",840 "TIMM/fbnetc_100",841 "TIMM/fbnetv3_b",842 "TIMM/gernet_l",843 "TIMM/ghostnet_100",844 "TIMM/gluon_inception_v3",845 "TIMM/hardcorenas_a",846 "TIMM/hrnet_w18",847 "TIMM/inception_v3",848 "TIMM/lcnet_050",849 "TIMM/levit_128",850 "TIMM/mnasnet_100",851 "TIMM/mobilenetv2_100",852 "TIMM/mobilenetv3_large_100",853 "TIMM/nfnet_l0",854 "TIMM/regnety_002",855 "TIMM/repvgg_a2",856 "TIMM/res2net50_14w_8s",857 "TIMM/res2next50",858 "TIMM/resmlp_12_224",859 "TIMM/resnet18",860 "TIMM/rexnet_100",861 "TIMM/selecsls42b",862 "TIMM/spnasnet_100",863 "TIMM/tf_efficientnet_b0",864 "TIMM/tinynet_a",865 "TIMM/visformer_small"866 ],867 "((T([16, 1000], f16), 1, False), {})": [868 "TIMM/nasnetalarge",869 "TIMM/pnasnet5large"870 ],871 "((T([16, 2], f16), 1, False), {})": [872 "HuggingFace/LayoutLMForSequenceClassification"873 ],874 "((T([2, 1000], f16), 1, False), {})": [875 "TIMM/cait_m36_384"876 ],877 "((T([2, 512], f16), 1, False), {})": [878 "HuggingFace/AlbertForQuestionAnswering"879 ],880 "((T([2048, 30522], f16), 1, False), {})": [881 "HuggingFace/DistilBertForMaskedLM",882 "HuggingFace/MobileBertForMaskedLM"883 ],884 "((T([2048, 32000], f16), 1, False), {})": [885 "HuggingFace/XLNetLMHeadModel"886 ],887 "((T([2048, 50005], f16), 1, False), {})": [888 "HuggingFace/PLBartForCausalLM"889 ],890 "((T([2048, 50265], f16), 1, False), {})": [891 "HuggingFace/BartForConditionalGeneration",892 "HuggingFace/DebertaForMaskedLM",893 "HuggingFace/MBartForCausalLM"894 ],895 "((T([254, 29056], f16), 1, False), {})": [896 "HuggingFace/MegatronBertForCausalLM"897 ],898 "((T([256, 128112], f16), 1, False), {})": [899 "HuggingFace/M2M100ForConditionalGeneration"900 ],901 "((T([256, 256008], f16), 1, False), {})": [902 "HuggingFace/XGLMForCausalLM"903 ],904 "((T([30, 3600], f16), -1, False), {})": [905 "TorchBench/pytorch_struct"906 ],907 "((T([30, 4771], f16, stride=(1, 30)), -1, False), {})": [908 "TorchBench/pytorch_struct"909 ],910 "((T([30], f16), -1, False), {})": [911 "TorchBench/pytorch_struct"912 ],913 "((T([32, 1000], f16), 1, False), {})": [914 "TIMM/convmixer_768_32",915 "TIMM/convnext_base",916 "TIMM/dpn107",917 "TIMM/gluon_senet154",918 "TIMM/gluon_xception65",919 "TIMM/legacy_senet154",920 "TIMM/resnest101e",921 "TIMM/swsl_resnext101_32x16d",922 "TIMM/twins_pcpvt_base"923 ],924 "((T([32, 128], f16), 1, False), {})": [925 "HuggingFace/DistilBertForQuestionAnswering",926 "HuggingFace/MobileBertForQuestionAnswering"927 ],928 "((T([4, 2], f16), 1, False), {})": [929 "HuggingFace/GPT2ForSequenceClassification"930 ],931 "((T([4, 512], f16), 1, False), {})": [932 "HuggingFace/DebertaForQuestionAnswering"933 ],934 "((T([4096, 50265], f16), 1, False), {})": [935 "HuggingFace/BartForCausalLM"936 ],937 "((T([508, 30522], f16), 1, False), {})": [938 "HuggingFace/RobertaForCausalLM"939 ],940 "((T([508, 50272], f16), 1, False), {})": [941 "HuggingFace/OPTForCausalLM"942 ],943 "((T([511, 30522], f16), 1, False), {})": [944 "HuggingFace/ElectraForCausalLM"945 ],946 "((T([511, 50257], f16), 1, False), {})": [947 "HuggingFace/DistillGPT2"948 ],949 "((T([512, 128100], f16), 1, False), {})": [950 "HuggingFace/DebertaV2ForMaskedLM"951 ],952 "((T([512, 30522], f16), 1, False), {})": [953 "HuggingFace/YituTechConvBert"954 ],955 "((T([512, 32000], f16), 1, False), {})": [956 "HuggingFace/GoogleFnet"957 ],958 "((T([512, 32005], f16), 1, False), {})": [959 "HuggingFace/CamemBert"960 ],961 "((T([512, 50265], f16), 1, False), {})": [962 "HuggingFace/PegasusForConditionalGeneration"963 ],964 "((T([64, 1000], f16), 1, False), {})": [965 "TIMM/beit_base_patch16_224",966 "TIMM/crossvit_9_240",967 "TIMM/cspdarknet53",968 "TIMM/deit_base_distilled_patch16_224",969 "TIMM/densenet121",970 "TIMM/dla102",971 "TIMM/ecaresnet101d",972 "TIMM/gmixer_24_224",973 "TIMM/gmlp_s16_224",974 "TIMM/jx_nest_base",975 "TIMM/mixer_b16_224",976 "TIMM/mixnet_l",977 "TIMM/mobilevit_s",978 "TIMM/pit_b_224",979 "TIMM/poolformer_m36",980 "TIMM/res2net101_26w_4s",981 "TIMM/sebotnet33ts_256",982 "TIMM/swin_base_patch4_window7_224",983 "TIMM/tf_mixnet_l",984 "TIMM/tnt_s_patch16_224",985 "TIMM/vit_base_patch16_224",986 "TIMM/volo_d1_224"987 ],988 "((T([64, 128], f16), 1, False), {})": [989 "HuggingFace/BertForQuestionAnswering",990 "HuggingFace/RobertaForQuestionAnswering"991 ],992 "((T([64, 512], f16), 1, False), {})": [993 "HuggingFace/ElectraForQuestionAnswering"994 ],995 "((T([8, 128], f16), 1, False), {})": [996 "HuggingFace/MegatronBertForQuestionAnswering"997 ],998 "((T([8192, 10000], f16), 1, False), {})": [999 "HuggingFace/Speech2Text2ForCausalLM"1000 ],1001 "((T([8192, 30522], f16), 1, False), {})": [1002 "HuggingFace/BertForMaskedLM",1003 "HuggingFace/LayoutLMForMaskedLM"1004 ],1005 "((T([8192, 50265], f16), 1, False), {})": [1006 "HuggingFace/BlenderbotSmallForCausalLM",1007 "HuggingFace/BlenderbotSmallForConditionalGeneration"1008 ]1009 },1010 "aten._log_softmax_backward_data.default": {1011 "((T([1, 2], f16), T([1, 2], f16), 1, f16), {})": [1012 "HuggingFace/GPTNeoForSequenceClassification"1013 ],1014 "((T([1, 512], f16), T([1, 512], f16), 1, f16), {})": [1015 "HuggingFace/DebertaV2ForQuestionAnswering"1016 ],1017 "((T([1024, 30000], f16), T([1024, 30000], f16), 1, f16), {})": [1018 "HuggingFace/AlbertForMaskedLM"1019 ],1020 "((T([1024, 50005], f16), T([1024, 50005], f16), 1, f16), {})": [1021 "HuggingFace/PLBartForConditionalGeneration"1022 ],1023 "((T([1024, 50265], f16), T([1024, 50265], f16), 1, f16), {})": [1024 "HuggingFace/AllenaiLongformerBase",1025 "HuggingFace/MBartForConditionalGeneration",1026 "HuggingFace/PegasusForCausalLM",1027 "HuggingFace/TrOCRForCausalLM"1028 ],1029 "((T([1024, 50358], f16), T([1024, 50358], f16), 1, f16), {})": [1030 "HuggingFace/BigBird"1031 ],1032 "((T([127, 50257], f32), T([127, 50257], f32), 1, f32), {})": [1033 "HuggingFace/GPTNeoForCausalLM"1034 ],1035 "((T([128, 1000], f16), T([128, 1000], f16), 1, f16), {})": [1036 "TIMM/adv_inception_v3",1037 "TIMM/botnet26t_256",1038 "TIMM/coat_lite_mini",1039 "TIMM/dm_nfnet_f0",1040 "TIMM/eca_botnext26ts_256",1041 "TIMM/eca_halonext26ts",1042 "TIMM/ese_vovnet19b_dw",1043 "TIMM/fbnetc_100",1044 "TIMM/fbnetv3_b",1045 "TIMM/gernet_l",1046 "TIMM/ghostnet_100",1047 "TIMM/gluon_inception_v3",1048 "TIMM/hardcorenas_a",1049 "TIMM/hrnet_w18",1050 "TIMM/inception_v3",1051 "TIMM/lcnet_050",1052 "TIMM/levit_128",1053 "TIMM/mnasnet_100",1054 "TIMM/mobilenetv2_100",1055 "TIMM/mobilenetv3_large_100",1056 "TIMM/nfnet_l0",1057 "TIMM/regnety_002",1058 "TIMM/repvgg_a2",1059 "TIMM/res2net50_14w_8s",1060 "TIMM/res2next50",1061 "TIMM/resmlp_12_224",1062 "TIMM/resnet18",1063 "TIMM/rexnet_100",1064 "TIMM/selecsls42b",1065 "TIMM/spnasnet_100",1066 "TIMM/tf_efficientnet_b0",1067 "TIMM/tinynet_a",1068 "TIMM/visformer_small"1069 ],1070 "((T([16, 1000], f16), T([16, 1000], f16), 1, f16), {})": [1071 "TIMM/nasnetalarge",1072 "TIMM/pnasnet5large"1073 ],1074 "((T([16, 2], f16), T([16, 2], f16), 1, f16), {})": [1075 "HuggingFace/LayoutLMForSequenceClassification"1076 ],1077 "((T([2, 1000], f16), T([2, 1000], f16), 1, f16), {})": [1078 "TIMM/cait_m36_384"1079 ],1080 "((T([2, 512], f16), T([2, 512], f16), 1, f16), {})": [1081 "HuggingFace/AlbertForQuestionAnswering"1082 ],1083 "((T([2048, 30522], f16), T([2048, 30522], f16), 1, f16), {})": [1084 "HuggingFace/DistilBertForMaskedLM",1085 "HuggingFace/MobileBertForMaskedLM"1086 ],1087 "((T([2048, 32000], f16), T([2048, 32000], f16), 1, f16), {})": [1088 "HuggingFace/XLNetLMHeadModel"1089 ],1090 "((T([2048, 50005], f16), T([2048, 50005], f16), 1, f16), {})": [1091 "HuggingFace/PLBartForCausalLM"1092 ],1093 "((T([2048, 50265], f16), T([2048, 50265], f16), 1, f16), {})": [1094 "HuggingFace/BartForConditionalGeneration",1095 "HuggingFace/DebertaForMaskedLM",1096 "HuggingFace/MBartForCausalLM"1097 ],1098 "((T([254, 29056], f16), T([254, 29056], f16), 1, f16), {})": [1099 "HuggingFace/MegatronBertForCausalLM"1100 ],1101 "((T([256, 128112], f16), T([256, 128112], f16), 1, f16), {})": [1102 "HuggingFace/M2M100ForConditionalGeneration"1103 ],1104 "((T([256, 256008], f16), T([256, 256008], f16), 1, f16), {})": [1105 "HuggingFace/XGLMForCausalLM"1106 ],1107 "((T([30, 3600], f16), T([30, 3600], f16), -1, f16), {})": [1108 "TorchBench/pytorch_struct"1109 ],1110 "((T([30, 4771], f16), T([30, 4771], f16), -1, f16), {})": [1111 "TorchBench/pytorch_struct"1112 ],1113 "((T([30], f16), T([30], f16), -1, f16), {})": [1114 "TorchBench/pytorch_struct"1115 ],1116 "((T([32, 1000], f16), T([32, 1000], f16), 1, f16), {})": [1117 "TIMM/convmixer_768_32",1118 "TIMM/convnext_base",1119 "TIMM/dpn107",1120 "TIMM/gluon_senet154",1121 "TIMM/gluon_xception65",1122 "TIMM/legacy_senet154",1123 "TIMM/resnest101e",1124 "TIMM/swsl_resnext101_32x16d",1125 "TIMM/twins_pcpvt_base"1126 ],1127 "((T([32, 128], f16), T([32, 128], f16), 1, f16), {})": [1128 "HuggingFace/DistilBertForQuestionAnswering",1129 "HuggingFace/MobileBertForQuestionAnswering"1130 ],1131 "((T([4, 2], f16), T([4, 2], f16), 1, f16), {})": [1132 "HuggingFace/GPT2ForSequenceClassification"1133 ],1134 "((T([4, 512], f16), T([4, 512], f16), 1, f16), {})": [1135 "HuggingFace/DebertaForQuestionAnswering"1136 ],1137 "((T([4096, 50265], f16), T([4096, 50265], f16), 1, f16), {})": [1138 "HuggingFace/BartForCausalLM"1139 ],1140 "((T([508, 30522], f16), T([508, 30522], f16), 1, f16), {})": [1141 "HuggingFace/RobertaForCausalLM"1142 ],1143 "((T([508, 50272], f16), T([508, 50272], f16), 1, f16), {})": [1144 "HuggingFace/OPTForCausalLM"1145 ],1146 "((T([511, 30522], f16), T([511, 30522], f16), 1, f16), {})": [1147 "HuggingFace/ElectraForCausalLM"1148 ],1149 "((T([511, 50257], f16), T([511, 50257], f16), 1, f16), {})": [1150 "HuggingFace/DistillGPT2"1151 ],1152 "((T([512, 128100], f16), T([512, 128100], f16), 1, f16), {})": [1153 "HuggingFace/DebertaV2ForMaskedLM"1154 ],1155 "((T([512, 30522], f16), T([512, 30522], f16), 1, f16), {})": [1156 "HuggingFace/YituTechConvBert"1157 ],1158 "((T([512, 32000], f16), T([512, 32000], f16), 1, f16), {})": [1159 "HuggingFace/GoogleFnet"1160 ],1161 "((T([512, 32005], f16), T([512, 32005], f16), 1, f16), {})": [1162 "HuggingFace/CamemBert"1163 ],1164 "((T([512, 50265], f16), T([512, 50265], f16), 1, f16), {})": [1165 "HuggingFace/PegasusForConditionalGeneration"1166 ],1167 "((T([64, 1000], f16), T([64, 1000], f16), 1, f16), {})": [1168 "TIMM/beit_base_patch16_224",1169 "TIMM/crossvit_9_240",1170 "TIMM/cspdarknet53",1171 "TIMM/deit_base_distilled_patch16_224",1172 "TIMM/densenet121",1173 "TIMM/dla102",1174 "TIMM/ecaresnet101d",1175 "TIMM/gmixer_24_224",1176 "TIMM/gmlp_s16_224",1177 "TIMM/jx_nest_base",1178 "TIMM/mixer_b16_224",1179 "TIMM/mixnet_l",1180 "TIMM/mobilevit_s",1181 "TIMM/pit_b_224",1182 "TIMM/poolformer_m36",1183 "TIMM/res2net101_26w_4s",1184 "TIMM/sebotnet33ts_256",1185 "TIMM/swin_base_patch4_window7_224",1186 "TIMM/tf_mixnet_l",1187 "TIMM/tnt_s_patch16_224",1188 "TIMM/vit_base_patch16_224",1189 "TIMM/volo_d1_224"1190 ],1191 "((T([64, 128], f16), T([64, 128], f16), 1, f16), {})": [1192 "HuggingFace/BertForQuestionAnswering",1193 "HuggingFace/RobertaForQuestionAnswering"1194 ],1195 "((T([64, 512], f16), T([64, 512], f16), 1, f16), {})": [1196 "HuggingFace/ElectraForQuestionAnswering"1197 ],1198 "((T([8, 128], f16), T([8, 128], f16), 1, f16), {})": [1199 "HuggingFace/MegatronBertForQuestionAnswering"1200 ],