Skip to content

Update channel_shuffle_op and transpose 2d to speed up ShuffleNet#9525

Closed
xiaomengy wants to merge 1 commit intopytorch:masterfrom
xiaomengy:export-D8889361
Closed

Update channel_shuffle_op and transpose 2d to speed up ShuffleNet#9525
xiaomengy wants to merge 1 commit intopytorch:masterfrom
xiaomengy:export-D8889361

Conversation

@xiaomengy
Copy link
Contributor

Summary: Update channel_shuffle_op and transpose 2d to speed up ShuffleNet

Differential Revision: D8889361

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

@xiaomengy xiaomengy requested a review from yinghai July 18, 2018 23:26

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

Copy link
Member

@houseroad houseroad left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The logic LGTM. Please make sure the iOS part is well tested.

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

2 similar comments
@bddppq
Copy link
Contributor

bddppq commented Jul 19, 2018

@pytorchbot retest this please

@houseroad
Copy link
Member

@pytorchbot retest this please

This comment was marked as off-topic.

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

@xiaomengy
Copy link
Contributor Author

Some benchmark: N = 128, C = 128, H = 128, W = 128, group = 4
NCHW: 212ms -> 84ms
NHWC: 2098ms -> 394ms

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

1 similar comment
@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

@houseroad
Copy link
Member

@pytorchbot retest this please

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

@xiaomengy
Copy link
Contributor Author

@pytorchbot retest this please

…torch#9525)

Summary:
Pull Request resolved: pytorch#9525

Update channel_shuffle_op and transpose 2d to speed up ShuffleNet

Reviewed By: houseroad

Differential Revision: D8889361

fbshipit-source-id: b5344955acb466e224e1c80cbaca721743c09e2f
@xiaomengy xiaomengy deleted the export-D8889361 branch July 21, 2018 21:41
jramseyer pushed a commit to jramseyer/pytorch that referenced this pull request Jul 30, 2018
…torch#9525)

Summary:
Pull Request resolved: pytorch#9525

Update channel_shuffle_op and transpose 2d to speed up ShuffleNet

Reviewed By: houseroad

Differential Revision: D8889361

fbshipit-source-id: 60196e819b6842becc53b4859b62d4419a0e2c6e
goodlux pushed a commit to goodlux/pytorch that referenced this pull request Aug 15, 2018
…torch#9525)

Summary:
Pull Request resolved: pytorch#9525

Update channel_shuffle_op and transpose 2d to speed up ShuffleNet

Reviewed By: houseroad

Differential Revision: D8889361

fbshipit-source-id: 60196e819b6842becc53b4859b62d4419a0e2c6e
@ezyang ezyang added the merged label Jun 26, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants