mindstudio-probe 1.0.3__py3-none-any.whl → 1.1.0__py3-none-any.whl

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 (278) hide show
  1. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/LICENSE +201 -201
  2. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/METADATA +36 -34
  3. mindstudio_probe-1.1.0.dist-info/RECORD +287 -0
  4. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/WHEEL +1 -1
  5. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/entry_points.txt +1 -0
  6. msprobe/README.md +131 -237
  7. msprobe/__init__.py +16 -1
  8. msprobe/{config/config.json → config.json} +47 -49
  9. msprobe/core/advisor/advisor.py +124 -124
  10. msprobe/core/advisor/advisor_const.py +58 -59
  11. msprobe/core/advisor/advisor_result.py +58 -58
  12. msprobe/core/common/const.py +402 -318
  13. msprobe/core/common/exceptions.py +99 -99
  14. msprobe/core/common/{file_check.py → file_utils.py} +523 -283
  15. msprobe/core/common/inplace_op_checker.py +38 -0
  16. msprobe/core/common/inplace_ops.yaml +251 -0
  17. msprobe/core/common/log.py +86 -69
  18. msprobe/core/common/utils.py +371 -616
  19. msprobe/core/common_config.py +78 -71
  20. msprobe/core/compare/acc_compare.py +472 -298
  21. msprobe/core/compare/check.py +180 -95
  22. msprobe/core/compare/compare_cli.py +69 -49
  23. msprobe/core/compare/highlight.py +259 -222
  24. msprobe/core/compare/multiprocessing_compute.py +174 -149
  25. msprobe/core/compare/npy_compare.py +310 -295
  26. msprobe/core/compare/utils.py +464 -429
  27. msprobe/core/data_dump/data_collector.py +153 -144
  28. msprobe/core/data_dump/data_processor/base.py +337 -293
  29. msprobe/core/data_dump/data_processor/factory.py +76 -59
  30. msprobe/core/data_dump/data_processor/mindspore_processor.py +192 -198
  31. msprobe/core/data_dump/data_processor/pytorch_processor.py +383 -389
  32. msprobe/core/data_dump/json_writer.py +117 -116
  33. msprobe/core/data_dump/scope.py +194 -178
  34. msprobe/core/grad_probe/constant.py +74 -70
  35. msprobe/core/grad_probe/grad_compare.py +170 -175
  36. msprobe/core/grad_probe/utils.py +77 -52
  37. msprobe/docs/01.installation.md +99 -0
  38. msprobe/docs/02.config_introduction.md +137 -0
  39. msprobe/docs/03.config_examples.md +237 -0
  40. msprobe/docs/04.acl_config_examples.md +78 -0
  41. msprobe/docs/05.data_dump_PyTorch.md +326 -0
  42. msprobe/docs/06.data_dump_MindSpore.md +285 -0
  43. msprobe/docs/07.accuracy_checker_PyTorch.md +297 -0
  44. msprobe/docs/08.accuracy_checker_online_PyTorch.md +238 -0
  45. msprobe/docs/09.accuracy_checker_MindSpore.md +68 -0
  46. msprobe/docs/10.accuracy_compare_PyTorch.md +327 -0
  47. msprobe/docs/11.accuracy_compare_MindSpore.md +333 -0
  48. msprobe/docs/12.overflow_check_PyTorch.md +79 -0
  49. msprobe/docs/13.overflow_check_MindSpore.md +31 -0
  50. msprobe/{pytorch/doc/parse_tool.md → docs/14.data_parse_PyTorch.md} +283 -286
  51. msprobe/docs/15.free_benchmarking_PyTorch.md +170 -0
  52. msprobe/docs/16.free_benchmarking_MindSpore.md +140 -0
  53. msprobe/{doc/grad_probe/grad_probe.md → docs/17.grad_probe.md} +205 -207
  54. msprobe/{pytorch/doc//321/205/320/254/320/270/321/207/342/225/221/342/224/220/321/207/342/226/223/342/225/233/321/205/342/225/221/320/266/321/206/320/277/320/244/321/205/320/277/342/225/243.md → docs/18.online_dispatch.md} +89 -90
  55. msprobe/docs/FAQ.md +189 -0
  56. msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md +146 -0
  57. msprobe/docs/img/free_benchmark_framework.png +0 -0
  58. msprobe/docs/img/ms_dump.png +0 -0
  59. msprobe/docs/img/ms_layer.png +0 -0
  60. msprobe/docs/img/pt_dump.png +0 -0
  61. msprobe/mindspore/__init__.py +2 -1
  62. msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +278 -245
  63. msprobe/mindspore/api_accuracy_checker/api_info.py +76 -69
  64. msprobe/mindspore/api_accuracy_checker/api_runner.py +155 -151
  65. msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py +196 -196
  66. msprobe/mindspore/api_accuracy_checker/cmd_parser.py +6 -0
  67. msprobe/mindspore/api_accuracy_checker/compute_element.py +238 -223
  68. msprobe/mindspore/api_accuracy_checker/main.py +8 -15
  69. msprobe/mindspore/api_accuracy_checker/type_mapping.py +113 -113
  70. msprobe/mindspore/api_accuracy_checker/utils.py +79 -62
  71. msprobe/mindspore/cell_processor.py +58 -34
  72. msprobe/mindspore/common/const.py +108 -87
  73. msprobe/mindspore/common/log.py +37 -37
  74. msprobe/mindspore/common/utils.py +97 -57
  75. msprobe/mindspore/compare/distributed_compare.py +62 -75
  76. msprobe/mindspore/compare/layer_mapping.py +146 -0
  77. msprobe/mindspore/compare/modify_mapping.py +107 -0
  78. msprobe/mindspore/compare/ms_compare.py +357 -117
  79. msprobe/mindspore/compare/ms_graph_compare.py +364 -317
  80. msprobe/mindspore/compare/ms_to_pt_api.yaml +399 -399
  81. msprobe/mindspore/debugger/debugger_config.py +69 -74
  82. msprobe/mindspore/debugger/precision_debugger.py +150 -107
  83. msprobe/mindspore/dump/dump_tool_factory.py +50 -35
  84. msprobe/mindspore/dump/hook_cell/api_registry.py +128 -104
  85. msprobe/mindspore/dump/hook_cell/hook_cell.py +55 -53
  86. msprobe/mindspore/dump/hook_cell/primitive_hooks.py +206 -0
  87. msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml +994 -925
  88. msprobe/mindspore/dump/hook_cell/wrap_api.py +121 -0
  89. msprobe/mindspore/dump/jit_dump.py +96 -56
  90. msprobe/mindspore/dump/kernel_graph_dump.py +75 -60
  91. msprobe/mindspore/dump/kernel_kbyk_dump.py +79 -65
  92. msprobe/mindspore/free_benchmark/api_pynative_self_check.py +131 -116
  93. msprobe/mindspore/free_benchmark/common/config.py +27 -12
  94. msprobe/mindspore/free_benchmark/common/handler_params.py +32 -17
  95. msprobe/mindspore/free_benchmark/common/utils.py +85 -71
  96. msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml +842 -842
  97. msprobe/mindspore/free_benchmark/decorator/dec_forward.py +57 -42
  98. msprobe/mindspore/free_benchmark/decorator/decorator_factory.py +122 -107
  99. msprobe/mindspore/free_benchmark/handler/base_handler.py +105 -90
  100. msprobe/mindspore/free_benchmark/handler/check_handler.py +56 -41
  101. msprobe/mindspore/free_benchmark/handler/fix_handler.py +51 -36
  102. msprobe/mindspore/free_benchmark/handler/handler_factory.py +36 -21
  103. msprobe/mindspore/free_benchmark/perturbation/add_noise.py +82 -67
  104. msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py +36 -21
  105. msprobe/mindspore/free_benchmark/perturbation/bit_noise.py +78 -63
  106. msprobe/mindspore/free_benchmark/perturbation/exchange_value.py +77 -0
  107. msprobe/mindspore/free_benchmark/perturbation/improve_precision.py +49 -34
  108. msprobe/mindspore/free_benchmark/perturbation/no_change.py +27 -12
  109. msprobe/mindspore/free_benchmark/perturbation/perturbation_factory.py +44 -27
  110. msprobe/mindspore/free_benchmark/self_check_tool_factory.py +48 -33
  111. msprobe/mindspore/grad_probe/global_context.py +100 -91
  112. msprobe/mindspore/grad_probe/grad_analyzer.py +231 -231
  113. msprobe/mindspore/grad_probe/grad_monitor.py +27 -27
  114. msprobe/mindspore/grad_probe/grad_stat_csv.py +131 -131
  115. msprobe/mindspore/grad_probe/hook.py +94 -92
  116. msprobe/mindspore/grad_probe/utils.py +29 -28
  117. msprobe/mindspore/ms_config.py +128 -126
  118. msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py +60 -45
  119. msprobe/mindspore/overflow_check/overflow_check_tool_factory.py +49 -34
  120. msprobe/mindspore/runtime.py +4 -4
  121. msprobe/mindspore/service.py +297 -354
  122. msprobe/mindspore/task_handler_factory.py +24 -24
  123. msprobe/msprobe.py +105 -107
  124. msprobe/pytorch/__init__.py +23 -4
  125. msprobe/pytorch/api_accuracy_checker/common/config.py +70 -55
  126. msprobe/pytorch/api_accuracy_checker/common/utils.py +246 -165
  127. msprobe/pytorch/api_accuracy_checker/compare/algorithm.py +230 -213
  128. msprobe/pytorch/api_accuracy_checker/compare/api_precision_compare.py +632 -581
  129. msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml +132 -132
  130. msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml +390 -390
  131. msprobe/pytorch/api_accuracy_checker/compare/compare.py +416 -381
  132. msprobe/pytorch/api_accuracy_checker/compare/compare_column.py +90 -73
  133. msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py +265 -244
  134. msprobe/pytorch/api_accuracy_checker/config.yaml +10 -10
  135. msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +370 -332
  136. msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py +221 -199
  137. msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py +150 -134
  138. msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py +518 -581
  139. msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py +213 -74
  140. msprobe/pytorch/api_accuracy_checker/run_ut/torch_ut_setting.json +7 -4
  141. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/attl.py +218 -202
  142. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/client.py +370 -324
  143. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py +227 -204
  144. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/dump_dispatch.py +110 -0
  145. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py +244 -218
  146. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/torch_ops_config.yaml +63 -0
  147. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py +44 -0
  148. msprobe/pytorch/bench_functions/__init__.py +30 -15
  149. msprobe/pytorch/bench_functions/apply_adam_w.py +43 -28
  150. msprobe/pytorch/bench_functions/confusion_transpose.py +34 -19
  151. msprobe/pytorch/bench_functions/fast_gelu.py +70 -55
  152. msprobe/pytorch/bench_functions/layer_norm_eval.py +21 -6
  153. msprobe/pytorch/bench_functions/linear.py +27 -12
  154. msprobe/pytorch/bench_functions/matmul_backward.py +63 -48
  155. msprobe/pytorch/bench_functions/npu_fusion_attention.py +538 -421
  156. msprobe/pytorch/bench_functions/rms_norm.py +30 -15
  157. msprobe/pytorch/bench_functions/rotary_mul.py +71 -52
  158. msprobe/pytorch/bench_functions/scaled_mask_softmax.py +41 -26
  159. msprobe/pytorch/bench_functions/swiglu.py +70 -55
  160. msprobe/pytorch/common/__init__.py +17 -2
  161. msprobe/pytorch/common/compare_script.template +14 -14
  162. msprobe/pytorch/common/log.py +33 -32
  163. msprobe/pytorch/common/parse_json.py +54 -39
  164. msprobe/pytorch/common/utils.py +310 -300
  165. msprobe/pytorch/compare/distributed_compare.py +66 -66
  166. msprobe/pytorch/compare/mapping.yaml +607 -607
  167. msprobe/pytorch/compare/match.py +49 -33
  168. msprobe/pytorch/compare/pt_compare.py +82 -40
  169. msprobe/pytorch/debugger/debugger_config.py +108 -95
  170. msprobe/pytorch/debugger/precision_debugger.py +173 -125
  171. msprobe/pytorch/free_benchmark/__init__.py +23 -8
  172. msprobe/pytorch/free_benchmark/common/constant.py +70 -70
  173. msprobe/pytorch/free_benchmark/common/counter.py +71 -71
  174. msprobe/pytorch/free_benchmark/common/enums.py +65 -37
  175. msprobe/pytorch/free_benchmark/common/params.py +144 -129
  176. msprobe/pytorch/free_benchmark/common/utils.py +118 -102
  177. msprobe/pytorch/free_benchmark/compare/grad_saver.py +200 -179
  178. msprobe/pytorch/free_benchmark/compare/single_benchmark.py +119 -104
  179. msprobe/pytorch/free_benchmark/main.py +120 -105
  180. msprobe/pytorch/free_benchmark/perturbed_layers/base_layer.py +28 -13
  181. msprobe/pytorch/free_benchmark/perturbed_layers/layer_factory.py +56 -41
  182. msprobe/pytorch/free_benchmark/perturbed_layers/npu/add_noise.py +105 -90
  183. msprobe/pytorch/free_benchmark/perturbed_layers/npu/bit_noise.py +119 -104
  184. msprobe/pytorch/free_benchmark/perturbed_layers/npu/change_value.py +87 -63
  185. msprobe/pytorch/free_benchmark/perturbed_layers/npu/improve_precision.py +83 -68
  186. msprobe/pytorch/free_benchmark/perturbed_layers/npu/no_change.py +43 -28
  187. msprobe/pytorch/free_benchmark/perturbed_layers/npu/npu_base_layser.py +60 -45
  188. msprobe/pytorch/free_benchmark/perturbed_layers/run_cpu.py +34 -19
  189. msprobe/pytorch/free_benchmark/result_handlers/base_handler.py +256 -217
  190. msprobe/pytorch/free_benchmark/result_handlers/check_handler.py +54 -39
  191. msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py +38 -23
  192. msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py +45 -30
  193. msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py +185 -170
  194. msprobe/pytorch/function_factory.py +91 -75
  195. msprobe/pytorch/functional/module_dump.py +84 -0
  196. msprobe/pytorch/grad_probe/grad_monitor.py +91 -90
  197. msprobe/pytorch/grad_probe/grad_stat_csv.py +128 -128
  198. msprobe/pytorch/hook_module/__init__.py +16 -1
  199. msprobe/pytorch/hook_module/api_registry.py +166 -161
  200. msprobe/pytorch/hook_module/hook_module.py +118 -120
  201. msprobe/pytorch/hook_module/support_wrap_ops.yaml +1879 -1877
  202. msprobe/pytorch/hook_module/utils.py +28 -29
  203. msprobe/pytorch/hook_module/wrap_aten.py +111 -110
  204. msprobe/pytorch/hook_module/wrap_distributed.py +77 -78
  205. msprobe/pytorch/hook_module/wrap_functional.py +104 -105
  206. msprobe/pytorch/hook_module/wrap_npu_custom.py +85 -84
  207. msprobe/pytorch/hook_module/wrap_tensor.py +69 -71
  208. msprobe/pytorch/hook_module/wrap_torch.py +84 -86
  209. msprobe/pytorch/hook_module/wrap_vf.py +60 -62
  210. msprobe/pytorch/module_processer.py +153 -138
  211. msprobe/pytorch/online_dispatch/__init__.py +20 -20
  212. msprobe/pytorch/online_dispatch/compare.py +235 -236
  213. msprobe/pytorch/online_dispatch/dispatch.py +271 -271
  214. msprobe/pytorch/online_dispatch/dump_compare.py +155 -156
  215. msprobe/pytorch/online_dispatch/single_compare.py +391 -391
  216. msprobe/pytorch/online_dispatch/torch_ops_config.yaml +57 -49
  217. msprobe/pytorch/online_dispatch/utils.py +127 -146
  218. msprobe/pytorch/parse.py +19 -4
  219. msprobe/pytorch/parse_tool/cli.py +31 -32
  220. msprobe/pytorch/parse_tool/lib/compare.py +259 -271
  221. msprobe/pytorch/parse_tool/lib/config.py +52 -52
  222. msprobe/pytorch/parse_tool/lib/file_desc.py +31 -31
  223. msprobe/pytorch/parse_tool/lib/interactive_cli.py +102 -102
  224. msprobe/pytorch/parse_tool/lib/parse_exception.py +54 -54
  225. msprobe/pytorch/parse_tool/lib/parse_tool.py +161 -158
  226. msprobe/pytorch/parse_tool/lib/utils.py +320 -321
  227. msprobe/pytorch/parse_tool/lib/visualization.py +85 -91
  228. msprobe/pytorch/pt_config.py +317 -187
  229. msprobe/pytorch/service.py +311 -252
  230. mindstudio_probe-1.0.3.dist-info/RECORD +0 -272
  231. msprobe/config/README.md +0 -539
  232. msprobe/mindspore/doc/compare.md +0 -58
  233. msprobe/mindspore/doc/dump.md +0 -217
  234. msprobe/mindspore/dump/hook_cell/wrap_functional.py +0 -91
  235. msprobe/mindspore/dump/hook_cell/wrap_tensor.py +0 -63
  236. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/ssl_config.py +0 -10
  237. msprobe/pytorch/doc/FAQ.md +0 -193
  238. msprobe/pytorch/doc/api_accuracy_checker.md +0 -313
  239. msprobe/pytorch/doc/api_accuracy_checker_online.md +0 -187
  240. msprobe/pytorch/doc/dump.md +0 -260
  241. msprobe/pytorch/doc/msprobe/321/207/342/226/223/342/225/233/321/205/342/225/221/320/266/321/205/342/225/226/320/265/321/205/320/225/342/225/226/321/206/320/245/342/226/221/321/206/320/235/320/276dump/321/206/320/260/320/227/321/205/320/227/320/226/321/206/320/220/320/267/321/210/320/223/342/225/234/321/205/320/257/342/225/221/321/207/342/225/221/342/224/220/321/206/320/232/320/265/321/205/320/241/320/232.md +0 -182
  242. msprobe/pytorch/doc/ptdbg_ascend_compare.md +0 -240
  243. msprobe/pytorch/doc/ptdbg_ascend_overview.md +0 -68
  244. msprobe/pytorch/doc/ptdbg_ascend_quickstart.md +0 -381
  245. msprobe/pytorch/doc/run_overflow_check.md +0 -25
  246. msprobe/pytorch/doc//321/206/320/247/320/260/321/206/320/260/320/227/321/206/320/255/320/226/321/205/342/225/226/320/265/321/205/320/225/342/225/226/321/205/320/254/342/225/221/321/206/320/251/320/277/321/211/320/272/320/234/321/210/320/277/320/221/321/205/320/242/320/234/321/206/320/220/320/267/321/210/320/223/342/225/234/321/205/320/257/342/225/221/321/207/342/225/221/342/224/220/321/206/320/232/320/265/321/205/320/241/320/232.md +0 -151
  247. msprobe/pytorch/functional/data_processor.py +0 -0
  248. msprobe/pytorch/functional/dump_module.py +0 -39
  249. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/top_level.txt +0 -0
  250. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_1.png +0 -0
  251. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_2.png +0 -0
  252. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_3.png +0 -0
  253. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_4.png +0 -0
  254. /msprobe/{pytorch/doc → docs}/img/GPT-3_1.png +0 -0
  255. /msprobe/{pytorch/doc → docs}/img/GPT-3_2.png +0 -0
  256. /msprobe/{pytorch/doc → docs}/img/GPT-3_3.png +0 -0
  257. /msprobe/{pytorch/doc → docs}/img/GPT-3_4.png +0 -0
  258. /msprobe/{pytorch/doc → docs}/img/GPT-3_5.png +0 -0
  259. /msprobe/{pytorch/doc → docs}/img/GPT-3_6.png +0 -0
  260. /msprobe/{pytorch/doc → docs}/img/GPT-3_7.png +0 -0
  261. /msprobe/{pytorch/doc → docs}/img/GPT-3_8.png +0 -0
  262. /msprobe/{pytorch/doc → docs}/img/YOLOV5S_1.png +0 -0
  263. /msprobe/{pytorch/doc → docs}/img/YOLOV5S_2.png +0 -0
  264. /msprobe/{pytorch/doc → docs}/img/accuracy_checking_details.png +0 -0
  265. /msprobe/{pytorch/doc → docs}/img/accuracy_checking_result.png +0 -0
  266. /msprobe/{pytorch/doc → docs}/img/api_precision_compare_details.png +0 -0
  267. /msprobe/{pytorch/doc → docs}/img/api_precision_compare_result.png +0 -0
  268. /msprobe/{pytorch/doc → docs}/img/auto_analyze_log.png +0 -0
  269. /msprobe/{pytorch/doc → docs}/img/compare_result_pkl.png +0 -0
  270. /msprobe/{pytorch/doc → docs}/img/compare_result_pkl_md5.png.png +0 -0
  271. /msprobe/{pytorch/doc → docs}/img/cpu_info.png +0 -0
  272. /msprobe/{config → docs}/img/free_benchmark.png +0 -0
  273. /msprobe/{doc/grad_probe/img/image-1.png → docs/img/grad_probe_image-1.png} +0 -0
  274. /msprobe/{doc/grad_probe/img/image-2.png → docs/img/grad_probe_image-2.png} +0 -0
  275. /msprobe/{doc/grad_probe/img/image-3.png → docs/img/grad_probe_image-3.png} +0 -0
  276. /msprobe/{doc/grad_probe/img/image-4.png → docs/img/grad_probe_image-4.png} +0 -0
  277. /msprobe/{doc/grad_probe/img/image.png → docs/img/grad_probe_image.png} +0 -0
  278. /msprobe/{pytorch/doc → docs}/img/module_compare.png +0 -0
msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml CHANGED
@@ -1,390 +1,390 @@
1
- mul:
2
- torch.float32:
3
- rtol: 1.0e-6
4
- small_value: 1.0e-6
5
- small_value_atol: 1.0e-9
6
- torch.float16:
7
- rtol: 1.0e-3
8
- small_value: 1.0e-3
9
- small_value_atol: 1.0e-5
10
- torch.bfloat16:
11
- rtol: 4.0e-3
12
- small_value: 1.0e-3
13
- small_value_atol: 1.0e-5
14
- mul_:
15
- torch.float32:
16
- rtol: 1.0e-6
17
- small_value: 1.0e-6
18
- small_value_atol: 1.0e-9
19
- torch.float16:
20
- rtol: 1.0e-3
21
- small_value: 1.0e-3
22
- small_value_atol: 1.0e-5
23
- torch.bfloat16:
24
- rtol: 4.0e-3
25
- small_value: 1.0e-3
26
- small_value_atol: 1.0e-5
27
- __mul__:
28
- torch.float32:
29
- rtol: 1.0e-6
30
- small_value: 1.0e-6
31
- small_value_atol: 1.0e-9
32
- torch.float16:
33
- rtol: 1.0e-3
34
- small_value: 1.0e-3
35
- small_value_atol: 1.0e-5
36
- torch.bfloat16:
37
- rtol: 4.0e-3
38
- small_value: 1.0e-3
39
- small_value_atol: 1.0e-5
40
- __imul__:
41
- torch.float32:
42
- rtol: 1.0e-6
43
- small_value: 1.0e-6
44
- small_value_atol: 1.0e-9
45
- torch.float16:
46
- rtol: 1.0e-3
47
- small_value: 1.0e-3
48
- small_value_atol: 1.0e-5
49
- torch.bfloat16:
50
- rtol: 4.0e-3
51
- small_value: 1.0e-3
52
- small_value_atol: 1.0e-5
53
- __rmul__:
54
- torch.float32:
55
- rtol: 1.0e-6
56
- small_value: 1.0e-6
57
- small_value_atol: 1.0e-9
58
- torch.float16:
59
- rtol: 1.0e-3
60
- small_value: 1.0e-3
61
- small_value_atol: 1.0e-5
62
- torch.bfloat16:
63
- rtol: 4.0e-3
64
- small_value: 1.0e-3
65
- small_value_atol: 1.0e-5
66
- add:
67
- torch.float32:
68
- rtol: 1.0e-6
69
- small_value: 1.0e-6
70
- small_value_atol: 1.0e-9
71
- torch.float16:
72
- rtol: 1.0e-3
73
- small_value: 1.0e-3
74
- small_value_atol: 1.0e-5
75
- torch.bfloat16:
76
- rtol: 4.0e-3
77
- small_value: 1.0e-3
78
- small_value_atol: 1.0e-5
79
- add_:
80
- torch.float32:
81
- rtol: 1.0e-6
82
- small_value: 1.0e-6
83
- small_value_atol: 1.0e-9
84
- torch.float16:
85
- rtol: 1.0e-3
86
- small_value: 1.0e-3
87
- small_value_atol: 1.0e-5
88
- torch.bfloat16:
89
- rtol: 4.0e-3
90
- small_value: 1.0e-3
91
- small_value_atol: 1.0e-5
92
- __add__:
93
- torch.float32:
94
- rtol: 1.0e-6
95
- small_value: 1.0e-6
96
- small_value_atol: 1.0e-9
97
- torch.float16:
98
- rtol: 1.0e-3
99
- small_value: 1.0e-3
100
- small_value_atol: 1.0e-5
101
- torch.bfloat16:
102
- rtol: 4.0e-3
103
- small_value: 1.0e-3
104
- small_value_atol: 1.0e-5
105
- __iadd__:
106
- torch.float32:
107
- rtol: 1.0e-6
108
- small_value: 1.0e-6
109
- small_value_atol: 1.0e-9
110
- torch.float16:
111
- rtol: 1.0e-3
112
- small_value: 1.0e-3
113
- small_value_atol: 1.0e-5
114
- torch.bfloat16:
115
- rtol: 4.0e-3
116
- small_value: 1.0e-3
117
- small_value_atol: 1.0e-5
118
- __radd__:
119
- torch.float32:
120
- rtol: 1.0e-6
121
- small_value: 1.0e-6
122
- small_value_atol: 1.0e-9
123
- torch.float16:
124
- rtol: 1.0e-3
125
- small_value: 1.0e-3
126
- small_value_atol: 1.0e-5
127
- torch.bfloat16:
128
- rtol: 4.0e-3
129
- small_value: 1.0e-3
130
- small_value_atol: 1.0e-5
131
- div:
132
- torch.float32:
133
- rtol: 1.0e-6
134
- small_value: 1.0e-6
135
- small_value_atol: 1.0e-9
136
- torch.float16:
137
- rtol: 1.0e-3
138
- small_value: 1.0e-3
139
- small_value_atol: 1.0e-5
140
- torch.bfloat16:
141
- rtol: 4.0e-3
142
- small_value: 1.0e-3
143
- small_value_atol: 1.0e-5
144
- div_:
145
- torch.float32:
146
- rtol: 1.0e-6
147
- small_value: 1.0e-6
148
- small_value_atol: 1.0e-9
149
- torch.float16:
150
- rtol: 1.0e-3
151
- small_value: 1.0e-3
152
- small_value_atol: 1.0e-5
153
- torch.bfloat16:
154
- rtol: 4.0e-3
155
- small_value: 1.0e-3
156
- small_value_atol: 1.0e-5
157
- __div__:
158
- torch.float32:
159
- rtol: 1.0e-6
160
- small_value: 1.0e-6
161
- small_value_atol: 1.0e-9
162
- torch.float16:
163
- rtol: 1.0e-3
164
- small_value: 1.0e-3
165
- small_value_atol: 1.0e-5
166
- torch.bfloat16:
167
- rtol: 4.0e-3
168
- small_value: 1.0e-3
169
- small_value_atol: 1.0e-5
170
- __idiv__:
171
- torch.float32:
172
- rtol: 1.0e-6
173
- small_value: 1.0e-6
174
- small_value_atol: 1.0e-9
175
- torch.float16:
176
- rtol: 1.0e-3
177
- small_value: 1.0e-3
178
- small_value_atol: 1.0e-5
179
- torch.bfloat16:
180
- rtol: 4.0e-3
181
- small_value: 1.0e-3
182
- small_value_atol: 1.0e-5
183
- divide:
184
- torch.float32:
185
- rtol: 1.0e-6
186
- small_value: 1.0e-6
187
- small_value_atol: 1.0e-9
188
- torch.float16:
189
- rtol: 1.0e-3
190
- small_value: 1.0e-3
191
- small_value_atol: 1.0e-5
192
- torch.bfloat16:
193
- rtol: 4.0e-3
194
- small_value: 1.0e-3
195
- small_value_atol: 1.0e-5
196
- divide_:
197
- torch.float32:
198
- rtol: 1.0e-6
199
- small_value: 1.0e-6
200
- small_value_atol: 1.0e-9
201
- torch.float16:
202
- rtol: 1.0e-3
203
- small_value: 1.0e-3
204
- small_value_atol: 1.0e-5
205
- torch.bfloat16:
206
- rtol: 4.0e-3
207
- small_value: 1.0e-3
208
- small_value_atol: 1.0e-5
209
- leaky_relu:
210
- torch.float32:
211
- rtol: 1.0e-6
212
- small_value: 1.0e-6
213
- small_value_atol: 1.0e-9
214
- torch.float16:
215
- rtol: 1.0e-3
216
- small_value: 1.0e-3
217
- small_value_atol: 1.0e-5
218
- torch.bfloat16:
219
- rtol: 4.0e-3
220
- small_value: 1.0e-3
221
- small_value_atol: 1.0e-5
222
- leaky_relu_:
223
- torch.float32:
224
- rtol: 1.0e-6
225
- small_value: 1.0e-6
226
- small_value_atol: 1.0e-9
227
- torch.float16:
228
- rtol: 1.0e-3
229
- small_value: 1.0e-3
230
- small_value_atol: 1.0e-5
231
- torch.bfloat16:
232
- rtol: 4.0e-3
233
- small_value: 1.0e-3
234
- small_value_atol: 1.0e-5
235
- prelu:
236
- torch.float32:
237
- rtol: 1.0e-6
238
- small_value: 1.0e-6
239
- small_value_atol: 1.0e-9
240
- torch.float16:
241
- rtol: 1.0e-3
242
- small_value: 1.0e-3
243
- small_value_atol: 1.0e-5
244
- torch.bfloat16:
245
- rtol: 4.0e-3
246
- small_value: 1.0e-3
247
- small_value_atol: 1.0e-5
248
- reciprocal:
249
- torch.float32:
250
- rtol: 1.0e-6
251
- small_value: 1.0e-6
252
- small_value_atol: 1.0e-9
253
- torch.float16:
254
- rtol: 1.0e-3
255
- small_value: 1.0e-3
256
- small_value_atol: 1.0e-5
257
- torch.bfloat16:
258
- rtol: 4.0e-3
259
- small_value: 1.0e-3
260
- small_value_atol: 1.0e-5
261
- reciprocal_:
262
- torch.float32:
263
- rtol: 1.0e-6
264
- small_value: 1.0e-6
265
- small_value_atol: 1.0e-9
266
- torch.float16:
267
- rtol: 1.0e-3
268
- small_value: 1.0e-3
269
- small_value_atol: 1.0e-5
270
- torch.bfloat16:
271
- rtol: 4.0e-3
272
- small_value: 1.0e-3
273
- small_value_atol: 1.0e-5
274
- rsqrt:
275
- torch.float32:
276
- rtol: 1.0e-6
277
- small_value: 1.0e-6
278
- small_value_atol: 1.0e-9
279
- torch.float16:
280
- rtol: 1.0e-3
281
- small_value: 1.0e-3
282
- small_value_atol: 1.0e-5
283
- torch.bfloat16:
284
- rtol: 4.0e-3
285
- small_value: 1.0e-3
286
- small_value_atol: 1.0e-5
287
- rsqrt_:
288
- torch.float32:
289
- rtol: 1.0e-6
290
- small_value: 1.0e-6
291
- small_value_atol: 1.0e-9
292
- torch.float16:
293
- rtol: 1.0e-3
294
- small_value: 1.0e-3
295
- small_value_atol: 1.0e-5
296
- torch.bfloat16:
297
- rtol: 4.0e-3
298
- small_value: 1.0e-3
299
- small_value_atol: 1.0e-5
300
- square:
301
- torch.float32:
302
- rtol: 1.0e-6
303
- small_value: 1.0e-6
304
- small_value_atol: 1.0e-9
305
- torch.float16:
306
- rtol: 1.0e-3
307
- small_value: 1.0e-3
308
- small_value_atol: 1.0e-5
309
- torch.bfloat16:
310
- rtol: 4.0e-3
311
- small_value: 1.0e-3
312
- small_value_atol: 1.0e-5
313
- square_:
314
- torch.float32:
315
- rtol: 1.0e-6
316
- small_value: 1.0e-6
317
- small_value_atol: 1.0e-9
318
- torch.float16:
319
- rtol: 1.0e-3
320
- small_value: 1.0e-3
321
- small_value_atol: 1.0e-5
322
- torch.bfloat16:
323
- rtol: 4.0e-3
324
- small_value: 1.0e-3
325
- small_value_atol: 1.0e-5
326
- sub:
327
- torch.float32:
328
- rtol: 1.0e-6
329
- small_value: 1.0e-6
330
- small_value_atol: 1.0e-9
331
- torch.float16:
332
- rtol: 1.0e-3
333
- small_value: 1.0e-3
334
- small_value_atol: 1.0e-5
335
- torch.bfloat16:
336
- rtol: 4.0e-3
337
- small_value: 1.0e-3
338
- small_value_atol: 1.0e-5
339
- sub_:
340
- torch.float32:
341
- rtol: 1.0e-6
342
- small_value: 1.0e-6
343
- small_value_atol: 1.0e-9
344
- torch.float16:
345
- rtol: 1.0e-3
346
- small_value: 1.0e-3
347
- small_value_atol: 1.0e-5
348
- torch.bfloat16:
349
- rtol: 4.0e-3
350
- small_value: 1.0e-3
351
- small_value_atol: 1.0e-5
352
- rsub:
353
- torch.float32:
354
- rtol: 1.0e-6
355
- small_value: 1.0e-6
356
- small_value_atol: 1.0e-9
357
- torch.float16:
358
- rtol: 1.0e-3
359
- small_value: 1.0e-3
360
- small_value_atol: 1.0e-5
361
- torch.bfloat16:
362
- rtol: 4.0e-3
363
- small_value: 1.0e-3
364
- small_value_atol: 1.0e-5
365
- __isub__:
366
- torch.float32:
367
- rtol: 1.0e-6
368
- small_value: 1.0e-6
369
- small_value_atol: 1.0e-9
370
- torch.float16:
371
- rtol: 1.0e-3
372
- small_value: 1.0e-3
373
- small_value_atol: 1.0e-5
374
- torch.bfloat16:
375
- rtol: 4.0e-3
376
- small_value: 1.0e-3
377
- small_value_atol: 1.0e-5
378
- __sub__:
379
- torch.float32:
380
- rtol: 1.0e-6
381
- small_value: 1.0e-6
382
- small_value_atol: 1.0e-9
383
- torch.float16:
384
- rtol: 1.0e-3
385
- small_value: 1.0e-3
386
- small_value_atol: 1.0e-5
387
- torch.bfloat16:
388
- rtol: 4.0e-3
389
- small_value: 1.0e-3
390
- small_value_atol: 1.0e-5
1
+ mul:
2
+ torch.float32:
3
+ rtol: 1.0e-6
4
+ small_value: 1.0e-6
5
+ small_value_atol: 1.0e-9
6
+ torch.float16:
7
+ rtol: 1.0e-3
8
+ small_value: 1.0e-3
9
+ small_value_atol: 1.0e-5
10
+ torch.bfloat16:
11
+ rtol: 4.0e-3
12
+ small_value: 1.0e-3
13
+ small_value_atol: 1.0e-5
14
+ mul_:
15
+ torch.float32:
16
+ rtol: 1.0e-6
17
+ small_value: 1.0e-6
18
+ small_value_atol: 1.0e-9
19
+ torch.float16:
20
+ rtol: 1.0e-3
21
+ small_value: 1.0e-3
22
+ small_value_atol: 1.0e-5
23
+ torch.bfloat16:
24
+ rtol: 4.0e-3
25
+ small_value: 1.0e-3
26
+ small_value_atol: 1.0e-5
27
+ __mul__:
28
+ torch.float32:
29
+ rtol: 1.0e-6
30
+ small_value: 1.0e-6
31
+ small_value_atol: 1.0e-9
32
+ torch.float16:
33
+ rtol: 1.0e-3
34
+ small_value: 1.0e-3
35
+ small_value_atol: 1.0e-5
36
+ torch.bfloat16:
37
+ rtol: 4.0e-3
38
+ small_value: 1.0e-3
39
+ small_value_atol: 1.0e-5
40
+ __imul__:
41
+ torch.float32:
42
+ rtol: 1.0e-6
43
+ small_value: 1.0e-6
44
+ small_value_atol: 1.0e-9
45
+ torch.float16:
46
+ rtol: 1.0e-3
47
+ small_value: 1.0e-3
48
+ small_value_atol: 1.0e-5
49
+ torch.bfloat16:
50
+ rtol: 4.0e-3
51
+ small_value: 1.0e-3
52
+ small_value_atol: 1.0e-5
53
+ __rmul__:
54
+ torch.float32:
55
+ rtol: 1.0e-6
56
+ small_value: 1.0e-6
57
+ small_value_atol: 1.0e-9
58
+ torch.float16:
59
+ rtol: 1.0e-3
60
+ small_value: 1.0e-3
61
+ small_value_atol: 1.0e-5
62
+ torch.bfloat16:
63
+ rtol: 4.0e-3
64
+ small_value: 1.0e-3
65
+ small_value_atol: 1.0e-5
66
+ add:
67
+ torch.float32:
68
+ rtol: 1.0e-6
69
+ small_value: 1.0e-6
70
+ small_value_atol: 1.0e-9
71
+ torch.float16:
72
+ rtol: 1.0e-3
73
+ small_value: 1.0e-3
74
+ small_value_atol: 1.0e-5
75
+ torch.bfloat16:
76
+ rtol: 4.0e-3
77
+ small_value: 1.0e-3
78
+ small_value_atol: 1.0e-5
79
+ add_:
80
+ torch.float32:
81
+ rtol: 1.0e-6
82
+ small_value: 1.0e-6
83
+ small_value_atol: 1.0e-9
84
+ torch.float16:
85
+ rtol: 1.0e-3
86
+ small_value: 1.0e-3
87
+ small_value_atol: 1.0e-5
88
+ torch.bfloat16:
89
+ rtol: 4.0e-3
90
+ small_value: 1.0e-3
91
+ small_value_atol: 1.0e-5
92
+ __add__:
93
+ torch.float32:
94
+ rtol: 1.0e-6
95
+ small_value: 1.0e-6
96
+ small_value_atol: 1.0e-9
97
+ torch.float16:
98
+ rtol: 1.0e-3
99
+ small_value: 1.0e-3
100
+ small_value_atol: 1.0e-5
101
+ torch.bfloat16:
102
+ rtol: 4.0e-3
103
+ small_value: 1.0e-3
104
+ small_value_atol: 1.0e-5
105
+ __iadd__:
106
+ torch.float32:
107
+ rtol: 1.0e-6
108
+ small_value: 1.0e-6
109
+ small_value_atol: 1.0e-9
110
+ torch.float16:
111
+ rtol: 1.0e-3
112
+ small_value: 1.0e-3
113
+ small_value_atol: 1.0e-5
114
+ torch.bfloat16:
115
+ rtol: 4.0e-3
116
+ small_value: 1.0e-3
117
+ small_value_atol: 1.0e-5
118
+ __radd__:
119
+ torch.float32:
120
+ rtol: 1.0e-6
121
+ small_value: 1.0e-6
122
+ small_value_atol: 1.0e-9
123
+ torch.float16:
124
+ rtol: 1.0e-3
125
+ small_value: 1.0e-3
126
+ small_value_atol: 1.0e-5
127
+ torch.bfloat16:
128
+ rtol: 4.0e-3
129
+ small_value: 1.0e-3
130
+ small_value_atol: 1.0e-5
131
+ div:
132
+ torch.float32:
133
+ rtol: 1.0e-6
134
+ small_value: 1.0e-6
135
+ small_value_atol: 1.0e-9
136
+ torch.float16:
137
+ rtol: 1.0e-3
138
+ small_value: 1.0e-3
139
+ small_value_atol: 1.0e-5
140
+ torch.bfloat16:
141
+ rtol: 4.0e-3
142
+ small_value: 1.0e-3
143
+ small_value_atol: 1.0e-5
144
+ div_:
145
+ torch.float32:
146
+ rtol: 1.0e-6
147
+ small_value: 1.0e-6
148
+ small_value_atol: 1.0e-9
149
+ torch.float16:
150
+ rtol: 1.0e-3
151
+ small_value: 1.0e-3
152
+ small_value_atol: 1.0e-5
153
+ torch.bfloat16:
154
+ rtol: 4.0e-3
155
+ small_value: 1.0e-3
156
+ small_value_atol: 1.0e-5
157
+ __div__:
158
+ torch.float32:
159
+ rtol: 1.0e-6
160
+ small_value: 1.0e-6
161
+ small_value_atol: 1.0e-9
162
+ torch.float16:
163
+ rtol: 1.0e-3
164
+ small_value: 1.0e-3
165
+ small_value_atol: 1.0e-5
166
+ torch.bfloat16:
167
+ rtol: 4.0e-3
168
+ small_value: 1.0e-3
169
+ small_value_atol: 1.0e-5
170
+ __idiv__:
171
+ torch.float32:
172
+ rtol: 1.0e-6
173
+ small_value: 1.0e-6
174
+ small_value_atol: 1.0e-9
175
+ torch.float16:
176
+ rtol: 1.0e-3
177
+ small_value: 1.0e-3
178
+ small_value_atol: 1.0e-5
179
+ torch.bfloat16:
180
+ rtol: 4.0e-3
181
+ small_value: 1.0e-3
182
+ small_value_atol: 1.0e-5
183
+ divide:
184
+ torch.float32:
185
+ rtol: 1.0e-6
186
+ small_value: 1.0e-6
187
+ small_value_atol: 1.0e-9
188
+ torch.float16:
189
+ rtol: 1.0e-3
190
+ small_value: 1.0e-3
191
+ small_value_atol: 1.0e-5
192
+ torch.bfloat16:
193
+ rtol: 4.0e-3
194
+ small_value: 1.0e-3
195
+ small_value_atol: 1.0e-5
196
+ divide_:
197
+ torch.float32:
198
+ rtol: 1.0e-6
199
+ small_value: 1.0e-6
200
+ small_value_atol: 1.0e-9
201
+ torch.float16:
202
+ rtol: 1.0e-3
203
+ small_value: 1.0e-3
204
+ small_value_atol: 1.0e-5
205
+ torch.bfloat16:
206
+ rtol: 4.0e-3
207
+ small_value: 1.0e-3
208
+ small_value_atol: 1.0e-5
209
+ leaky_relu:
210
+ torch.float32:
211
+ rtol: 1.0e-6
212
+ small_value: 1.0e-6
213
+ small_value_atol: 1.0e-9
214
+ torch.float16:
215
+ rtol: 1.0e-3
216
+ small_value: 1.0e-3
217
+ small_value_atol: 1.0e-5
218
+ torch.bfloat16:
219
+ rtol: 4.0e-3
220
+ small_value: 1.0e-3
221
+ small_value_atol: 1.0e-5
222
+ leaky_relu_:
223
+ torch.float32:
224
+ rtol: 1.0e-6
225
+ small_value: 1.0e-6
226
+ small_value_atol: 1.0e-9
227
+ torch.float16:
228
+ rtol: 1.0e-3
229
+ small_value: 1.0e-3
230
+ small_value_atol: 1.0e-5
231
+ torch.bfloat16:
232
+ rtol: 4.0e-3
233
+ small_value: 1.0e-3
234
+ small_value_atol: 1.0e-5
235
+ prelu:
236
+ torch.float32:
237
+ rtol: 1.0e-6
238
+ small_value: 1.0e-6
239
+ small_value_atol: 1.0e-9
240
+ torch.float16:
241
+ rtol: 1.0e-3
242
+ small_value: 1.0e-3
243
+ small_value_atol: 1.0e-5
244
+ torch.bfloat16:
245
+ rtol: 4.0e-3
246
+ small_value: 1.0e-3
247
+ small_value_atol: 1.0e-5
248
+ reciprocal:
249
+ torch.float32:
250
+ rtol: 1.0e-6
251
+ small_value: 1.0e-6
252
+ small_value_atol: 1.0e-9
253
+ torch.float16:
254
+ rtol: 1.0e-3
255
+ small_value: 1.0e-3
256
+ small_value_atol: 1.0e-5
257
+ torch.bfloat16:
258
+ rtol: 4.0e-3
259
+ small_value: 1.0e-3
260
+ small_value_atol: 1.0e-5
261
+ reciprocal_:
262
+ torch.float32:
263
+ rtol: 1.0e-6
264
+ small_value: 1.0e-6
265
+ small_value_atol: 1.0e-9
266
+ torch.float16:
267
+ rtol: 1.0e-3
268
+ small_value: 1.0e-3
269
+ small_value_atol: 1.0e-5
270
+ torch.bfloat16:
271
+ rtol: 4.0e-3
272
+ small_value: 1.0e-3
273
+ small_value_atol: 1.0e-5
274
+ rsqrt:
275
+ torch.float32:
276
+ rtol: 1.0e-6
277
+ small_value: 1.0e-6
278
+ small_value_atol: 1.0e-9
279
+ torch.float16:
280
+ rtol: 1.0e-3
281
+ small_value: 1.0e-3
282
+ small_value_atol: 1.0e-5
283
+ torch.bfloat16:
284
+ rtol: 4.0e-3
285
+ small_value: 1.0e-3
286
+ small_value_atol: 1.0e-5
287
+ rsqrt_:
288
+ torch.float32:
289
+ rtol: 1.0e-6
290
+ small_value: 1.0e-6
291
+ small_value_atol: 1.0e-9
292
+ torch.float16:
293
+ rtol: 1.0e-3
294
+ small_value: 1.0e-3
295
+ small_value_atol: 1.0e-5
296
+ torch.bfloat16:
297
+ rtol: 4.0e-3
298
+ small_value: 1.0e-3
299
+ small_value_atol: 1.0e-5
300
+ square:
301
+ torch.float32:
302
+ rtol: 1.0e-6
303
+ small_value: 1.0e-6
304
+ small_value_atol: 1.0e-9
305
+ torch.float16:
306
+ rtol: 1.0e-3
307
+ small_value: 1.0e-3
308
+ small_value_atol: 1.0e-5
309
+ torch.bfloat16:
310
+ rtol: 4.0e-3
311
+ small_value: 1.0e-3
312
+ small_value_atol: 1.0e-5
313
+ square_:
314
+ torch.float32:
315
+ rtol: 1.0e-6
316
+ small_value: 1.0e-6
317
+ small_value_atol: 1.0e-9
318
+ torch.float16:
319
+ rtol: 1.0e-3
320
+ small_value: 1.0e-3
321
+ small_value_atol: 1.0e-5
322
+ torch.bfloat16:
323
+ rtol: 4.0e-3
324
+ small_value: 1.0e-3
325
+ small_value_atol: 1.0e-5
326
+ sub:
327
+ torch.float32:
328
+ rtol: 1.0e-6
329
+ small_value: 1.0e-6
330
+ small_value_atol: 1.0e-9
331
+ torch.float16:
332
+ rtol: 1.0e-3
333
+ small_value: 1.0e-3
334
+ small_value_atol: 1.0e-5
335
+ torch.bfloat16:
336
+ rtol: 4.0e-3
337
+ small_value: 1.0e-3
338
+ small_value_atol: 1.0e-5
339
+ sub_:
340
+ torch.float32:
341
+ rtol: 1.0e-6
342
+ small_value: 1.0e-6
343
+ small_value_atol: 1.0e-9
344
+ torch.float16:
345
+ rtol: 1.0e-3
346
+ small_value: 1.0e-3
347
+ small_value_atol: 1.0e-5
348
+ torch.bfloat16:
349
+ rtol: 4.0e-3
350
+ small_value: 1.0e-3
351
+ small_value_atol: 1.0e-5
352
+ rsub:
353
+ torch.float32:
354
+ rtol: 1.0e-6
355
+ small_value: 1.0e-6
356
+ small_value_atol: 1.0e-9
357
+ torch.float16:
358
+ rtol: 1.0e-3
359
+ small_value: 1.0e-3
360
+ small_value_atol: 1.0e-5
361
+ torch.bfloat16:
362
+ rtol: 4.0e-3
363
+ small_value: 1.0e-3
364
+ small_value_atol: 1.0e-5
365
+ __isub__:
366
+ torch.float32:
367
+ rtol: 1.0e-6
368
+ small_value: 1.0e-6
369
+ small_value_atol: 1.0e-9
370
+ torch.float16:
371
+ rtol: 1.0e-3
372
+ small_value: 1.0e-3
373
+ small_value_atol: 1.0e-5
374
+ torch.bfloat16:
375
+ rtol: 4.0e-3
376
+ small_value: 1.0e-3
377
+ small_value_atol: 1.0e-5
378
+ __sub__:
379
+ torch.float32:
380
+ rtol: 1.0e-6
381
+ small_value: 1.0e-6
382
+ small_value_atol: 1.0e-9
383
+ torch.float16:
384
+ rtol: 1.0e-3
385
+ small_value: 1.0e-3
386
+ small_value_atol: 1.0e-5
387
+ torch.bfloat16:
388
+ rtol: 4.0e-3
389
+ small_value: 1.0e-3
390
+ small_value_atol: 1.0e-5