Skip to content

A
akg

MindSpore / akg

通知 58
Star 7
Fork 7
A
akg
提交 bf3a6c3a 编写于 作者: M mindspore-ci-bot 提交者： Gitee
浏览文件
操作

!96 [CI] adapt resnet50 case for profiling test 

Merge pull request !96 from brovensmile/master
上级 11ed37cc 6c96b01d
隐藏空白更改
内联 并排
Showing with 48 addition and 48 deletion
tests/perf_benchmark/test_resnet50_all_001.py
浏览文件 @ bf3a6c3a
......@@ -917,153 +917,153 @@ class TestResnet50_001(BaseCaseRun):
# bn_split
("test_resnet50_bn_split_000", bn_split_run,
((32, 128, 7, 7, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 128, 7, 7, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_000"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_001", bn_split_run,
((32, 16, 14, 14, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 16, 14, 14, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_001"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_002", bn_split_run,
((32, 16, 56, 56, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 16, 56, 56, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_002"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_003", bn_split_run,
((32, 32, 28, 28, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 32, 28, 28, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_003"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_004", bn_split_run,
((32, 32, 7, 7, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 32, 7, 7, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_004"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_005", bn_split_run,
((32, 4, 112, 112, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 4, 112, 112, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_005"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_006", bn_split_run,
((32, 4, 56, 56, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 4, 56, 56, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_006"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_007", bn_split_run,
((32, 64, 14, 14, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 64, 14, 14, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_007"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_008", bn_split_run,
((32, 8, 28, 28, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 8, 28, 28, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_008"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_009", bn_split_run,
((32, 8, 56, 56, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 8, 56, 56, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_009"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_010", bn_split_run,
((32, 16, 28, 28, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 16, 28, 28, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_010"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_011", bn_split_run,
((32, 32, 14, 14, 16), "float32", 0.1, 1e-4, "resnet50_bn_split"),
((32, 32, 14, 14, 16), "float32", 0.1, 1e-4, "resnet50_bn_split_011"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_012", bn_split_run,
((32, 128, 7, 7, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 128, 7, 7, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_012"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_013", bn_split_run,
((32, 16, 14, 14, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 16, 14, 14, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_013"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_014", bn_split_run,
((32, 16, 56, 56, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 16, 56, 56, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_014"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_015", bn_split_run,
((32, 32, 28, 28, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 32, 28, 28, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_015"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_016", bn_split_run,
((32, 32, 7, 7, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 32, 7, 7, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_016"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_017", bn_split_run,
((32, 4, 112, 112, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 4, 112, 112, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_017"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_018", bn_split_run,
((32, 4, 56, 56, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 4, 56, 56, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_018"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_019", bn_split_run,
((32, 64, 14, 14, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 64, 14, 14, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_019"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_020", bn_split_run,
((32, 8, 28, 28, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 8, 28, 28, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_020"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_021", bn_split_run,
((32, 8, 56, 56, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 8, 56, 56, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_021"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_022", bn_split_run,
((32, 16, 28, 28, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 16, 28, 28, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_022"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_bn_split_023", bn_split_run,
((32, 32, 14, 14, 16), "float16", 0.1, 1e-4, "resnet50_bn_split"),
((32, 32, 14, 14, 16), "float16", 0.1, 1e-4, "resnet50_bn_split_023"),
["level0", "rpc", "rpc_cloud"]),
# fused_bn_grad_5D_all
("test_resnet50_fused_bn_grad_5D_all_000", fused_bn_grad_5D_all_run,
((32, 128, 7, 7, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 128, 7, 7, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_000"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_001", fused_bn_grad_5D_all_run,
((32, 16, 14, 14, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 16, 14, 14, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_001"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_002", fused_bn_grad_5D_all_run,
((32, 16, 56, 56, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 16, 56, 56, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_002"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_003", fused_bn_grad_5D_all_run,
((32, 32, 28, 28, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 32, 28, 28, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_003"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_004", fused_bn_grad_5D_all_run,
((32, 32, 7, 7, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 32, 7, 7, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_004"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_005", fused_bn_grad_5D_all_run,
((32, 4, 112, 112, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 4, 112, 112, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_005"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_006", fused_bn_grad_5D_all_run,
((32, 4, 56, 56, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 4, 56, 56, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_006"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_007", fused_bn_grad_5D_all_run,
((32, 64, 14, 14, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 64, 14, 14, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_007"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_008", fused_bn_grad_5D_all_run,
((32, 8, 28, 28, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 8, 28, 28, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_008"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_009", fused_bn_grad_5D_all_run,
((32, 8, 56, 56, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_010"),
((32, 8, 56, 56, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_009"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_010", fused_bn_grad_5D_all_run,
((32, 16, 28, 28, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_011"),
((32, 16, 28, 28, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_010"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_011", fused_bn_grad_5D_all_run,
((32, 32, 14, 14, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_012"),
((32, 32, 14, 14, 16), "float32", 1e-4, "resnet50_fused_bn_grad_5D_all_011"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_012", fused_bn_grad_5D_all_run,
((32, 128, 7, 7, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 128, 7, 7, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_012"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_013", fused_bn_grad_5D_all_run,
((32, 16, 14, 14, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 16, 14, 14, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_013"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_014", fused_bn_grad_5D_all_run,
((32, 16, 56, 56, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 16, 56, 56, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_014"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_015", fused_bn_grad_5D_all_run,
((32, 32, 28, 28, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 32, 28, 28, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_015"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_016", fused_bn_grad_5D_all_run,
((32, 32, 7, 7, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 32, 7, 7, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_016"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_017", fused_bn_grad_5D_all_run,
((32, 4, 112, 112, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 4, 112, 112, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_017"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_018", fused_bn_grad_5D_all_run,
((32, 4, 56, 56, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 4, 56, 56, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_018"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_019", fused_bn_grad_5D_all_run,
((32, 64, 14, 14, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 64, 14, 14, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_019"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_020", fused_bn_grad_5D_all_run,
((32, 8, 28, 28, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_01"),
((32, 8, 28, 28, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_020"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_021", fused_bn_grad_5D_all_run,
((32, 8, 56, 56, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_010"),
((32, 8, 56, 56, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_021"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_022", fused_bn_grad_5D_all_run,
((32, 16, 28, 28, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_011"),
((32, 16, 28, 28, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_022"),
["level0", "rpc", "rpc_cloud"]),
("test_resnet50_fused_bn_grad_5D_all_023", fused_bn_grad_5D_all_run,
((32, 32, 14, 14, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_012"),
((32, 32, 14, 14, 16), "float16", 1e-4, "resnet50_fused_bn_grad_5D_all_023"),
["level0", "rpc", "rpc_cloud"]),
# maxpool_grad_with_argmax
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或