Skip to content

Commit

Permalink
Merge pull request #1958 from StanfordAHA/maxpool_skip_pnr_search
Browse files Browse the repository at this point in the history
make maxpooling smaller; put it forward in regression
  • Loading branch information
yuchen-mei authored Nov 19, 2024
2 parents 4c3261d + 8c42f8d commit 129bebe
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 9 deletions.
12 changes: 6 additions & 6 deletions aha/util/application_parameters.json
Original file line number Diff line number Diff line change
Expand Up @@ -122,37 +122,37 @@
},
"apps/maxpooling":{
"default":{
"HALIDE_GEN_ARGS":"in_img=114 ksize=3 stride=2 n_ic=112 unroll=14",
"HALIDE_GEN_ARGS":"in_img=58 ksize=3 stride=2 n_ic=56 unroll=7",
"HL_TARGET":"host-x86-64",
"DISABLE_GP":"1",
"PNR_PLACER_EXP":"15"
},
"fastest":{
"HALIDE_GEN_ARGS":"in_img=114 ksize=3 stride=2 n_ic=112 unroll=14",
"HALIDE_GEN_ARGS":"in_img=58 ksize=3 stride=2 n_ic=56 unroll=7",
"HL_TARGET":"host-x86-64",
"DISABLE_GP":"1"
},
"sweep":{
"HALIDE_GEN_ARGS":"in_img=114 ksize=3 stride=2 n_ic=112 unroll=14",
"HALIDE_GEN_ARGS":"in_img=58 ksize=3 stride=2 n_ic=56 unroll=7",
"HL_TARGET":"host-x86-64",
"DISABLE_GP":"1",
"SWEEP_PNR_PLACER_EXP":"1"
}
},
"apps/maxpooling_fp":{
"default":{
"HALIDE_GEN_ARGS":"in_img=57 ksize=3 stride=2 n_ic=56 unroll=14",
"HALIDE_GEN_ARGS":"in_img=29 ksize=3 stride=2 n_ic=28 unroll=7",
"HL_TARGET":"host-x86-64-bfloat_hardware",
"DISABLE_GP":"1",
"PNR_PLACER_EXP":"6"
},
"fastest":{
"HALIDE_GEN_ARGS":"in_img=57 ksize=3 stride=2 n_ic=56 unroll=14",
"HALIDE_GEN_ARGS":"in_img=29 ksize=3 stride=2 n_ic=28 unroll=7",
"HL_TARGET":"host-x86-64-bfloat_hardware",
"DISABLE_GP":"1"
},
"sweep":{
"HALIDE_GEN_ARGS":"in_img=57 ksize=3 stride=2 n_ic=56 unroll=14",
"HALIDE_GEN_ARGS":"in_img=29 ksize=3 stride=2 n_ic=28 unroll=7",
"HL_TARGET":"host-x86-64-bfloat_hardware",
"DISABLE_GP":"1",
"SWEEP_PNR_PLACER_EXP":"1"
Expand Down
5 changes: 3 additions & 2 deletions aha/util/regress_tests/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@ def __init__(self, testname):
"tensor3_ttv",
]
glb_tests = [
"apps/maxpooling",
"apps/pointwise",
"apps/gaussian",
"apps/camera_pipeline_2x2",
Expand Down Expand Up @@ -207,6 +208,7 @@ def __init__(self, testname):

]
glb_tests = [
"apps/maxpooling",
"apps/pointwise",
"tests/rom",
"tests/arith",
Expand All @@ -231,15 +233,14 @@ def __init__(self, testname):
"apps/unsharp",
"apps/harris_color",
"apps/camera_pipeline_2x2",
"apps/maxpooling",
"apps/matrix_multiplication"
]
glb_tests_fp = [
"apps/maxpooling_fp",
"tests/fp_pointwise",
"tests/fp_arith",
"tests/fp_comp",
"tests/fp_conv_7_7",
"apps/maxpooling_fp",
"apps/matrix_multiplication_fp",
"apps/mcunet_in_sequential_0_fp",
"apps/depthwise_conv_stream_fp",
Expand Down
2 changes: 1 addition & 1 deletion garnet

0 comments on commit 129bebe

Please sign in to comment.