Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Improve NHWC depthwise convolution for AArch64 #6095

Merged
merged 11 commits into from
Aug 13, 2020
Merged

Improve NHWC depthwise convolution for AArch64 #6095

merged 11 commits into from
Aug 13, 2020

Commits on Aug 12, 2020

  1. Improve NHWC depthwise convolution for aarch64 

    We created a default schedule (no auto-tuning or tensorization) named
depthwise_conv2d_nhwc which does a decent job at optimizing depthwise
for NHWC layouts (on aarch64).

Change-Id: I01e32903f6c1950623f33eae18484e70244fe0af
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    407bd6d View commit details
    Browse the repository at this point in the history

  2. Add tuning knobs in depthwise schedule 

    Change-Id: I15080e7f12b16e6c6aba99a04e42023845eeabf1
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    0385a61 View commit details
    Browse the repository at this point in the history

  3. Introduce padding policy 

    Change-Id: If12a6d05dce9153861550ddef1ee5216809dd1e1
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    edc6c1b View commit details
    Browse the repository at this point in the history

  4. Vectorize padding 

    Change-Id: I7e2062a40358bf111c0366a449945eb077fb2e30
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    a317d76 View commit details
    Browse the repository at this point in the history

  5. Legalize depthwise convolution (2x improvement) and fix tuning issue 

    Change-Id: I4b82c58b167e40b0b7747d28293bbb488c505dd9
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    30a1204 View commit details
    Browse the repository at this point in the history

  6. Adding assert on padding 

    Change-Id: Idf8eeaaface5eb7799109cd00f437e404778b9cd
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    bf976b4 View commit details
    Browse the repository at this point in the history

  7. Fix python linting 

    Change-Id: Iac16a8daea1268f0eb331fe4ec18a62408106cf9
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    69d1f11 View commit details
    Browse the repository at this point in the history

  8. Removing commented code 

    Change-Id: I1412f22ad9864273d77a7bf38a6768694339b7f0
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    c53865d View commit details
    Browse the repository at this point in the history

  9. Revert test file to make CI pass 

    Change-Id: Ica3eff8f9f0fd4c6f32f7ae80adc922f8b16cec9
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    849e280 View commit details
    Browse the repository at this point in the history

  10. Enabling only arm_cpu tests 

    Change-Id: Icbaafcb39e892a5d1a4685133c1699e4d1a8e07e
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    ec13395 View commit details
    Browse the repository at this point in the history

  11. Rebasing 

    Change-Id: Ibb23f1d4e0d0107e4e3b3571437161cdc2ee2909
    Giuseppe Rossini committed Aug 12, 2020
    Configuration menu
    Copy the full SHA
    6ef6572 View commit details
    Browse the repository at this point in the history