Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[Backport] Enabling BRGEMM FullyConnected based on shapes #20568

Merged
merged 3 commits into from
Sep 6, 2021
Merged

[Backport] Enabling BRGEMM FullyConnected based on shapes #20568

merged 3 commits into from
Sep 6, 2021

Commits on Sep 3, 2021

  1. [v1.x][Feature] Add flag for disabling oneDNN BRGEMM implementation o… 

    …f FC (apache#20450)

* Add flag for disabling oneDNN BRGEMM implementation of FC

* Review fixes

* Update env_var.md
    @bgawrych
    bgawrych committed Sep 3, 2021
    Configuration menu
    Copy the full SHA
    17a2af0 View commit details
    Browse the repository at this point in the history

  2. [v1.x] Enabling BRGEMM FullyConnected based on shapes (apache#20533) 

    * Enable brgemm based on input info

* fix sanity

* Review fixes

* Change function name

* Fix typo

* Align variable assignments

* Fix review

* use const reference
    @bgawrych
    bgawrych committed Sep 3, 2021
    Configuration menu
    Copy the full SHA
    dd70522 View commit details
    Browse the repository at this point in the history

  3. Update flag name

    @bgawrych
    bgawrych committed Sep 3, 2021
    Configuration menu
    Copy the full SHA
    fb7be33 View commit details
    Browse the repository at this point in the history