Skip to content

[TOPI] Memoize winograd matrix - #3687

Merged
tqchen merged 3 commits into
apache:masterfrom
merrymercy:pr-memorize-winograd
Aug 2, 2019
Merged

[TOPI] Memoize winograd matrix#3687
tqchen merged 3 commits into
apache:masterfrom
merrymercy:pr-memorize-winograd

Conversation

@merrymercy

@merrymercymerrymercy commented Aug 1, 2019

Copy link
Copy Markdown
Member

I found #3553 makes feature extraction for arm cpu about 4x slower, which will cause tuning about 2x slower.

Memoizing winograd matrix computation can fix this performance regression, and is slightly faster the the original hard-coded one.

@merrymercy

Copy link
Copy Markdown
MemberAuthor

cc @zlateski@cbalint13

@merrymercy

Copy link
Copy Markdown
MemberAuthor

@cbalint13 please approve

@cbalint13

Copy link
Copy Markdown
Contributor

Looks very good to me.

Comment threadtopi/python/topi/nn/winograd_util.py
@tqchentqchen added the status: need update need update based on feedbacks label Aug 1, 2019
@tqchen

Copy link
Copy Markdown
Member

@merrymercy please act on @cbalint13 's comment

@merrymercy

Copy link
Copy Markdown
MemberAuthor

It is resolved.

@tqchen
tqchen merged commit 7de8a3a into apache:masterAug 2, 2019
@tqchentqchen added status: accepted and removed status: need update need update based on feedbacks labels Aug 2, 2019
@tqchen

Copy link
Copy Markdown
Member

Thanks @cbalint13@merrymercy !

@merrymercy
merrymercy deleted the pr-memorize-winograd branch August 3, 2019 12:11
wweic pushed a commit to wweic/tvm that referenced this pull request Aug 9, 2019
* [TOPI] Memoize winograd matrix
* lint
* Fix name
wweic pushed a commit to neo-ai/tvm that referenced this pull request Sep 6, 2019
* [TOPI] Memoize winograd matrix
* lint
* Fix name
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants

@merrymercy@cbalint13@tqchen