Skip to content

[Metal] Fix_conv_activation_pass and add MPS of Relu6 HardSigmoid HardSwish#7662

Merged
zhangjun merged 2 commits intoPaddlePaddle:developfrom
xiaoxiaohehe001:add_convactivate_fuse
Nov 22, 2021
Merged

[Metal] Fix_conv_activation_pass and add MPS of Relu6 HardSigmoid HardSwish#7662
zhangjun merged 2 commits intoPaddlePaddle:developfrom
xiaoxiaohehe001:add_convactivate_fuse

Conversation

@xiaoxiaohehe001
Copy link
Collaborator

@xiaoxiaohehe001 xiaoxiaohehe001 commented Nov 17, 2021

增加了RELU6 Hardsigmoid等MPS实现,增加了HardSwish非MPS实现
MobileNetV2 模型 MPS 测试

MobileNetV2 模型 MPS 测试

模型 cost(ms) 设备 MPS
MobileNetV2 10.3771 iPhone11 metal YES Relu6
MobileNetV2 11.4458 iPhone11 metal NO Relu6

有显著加速效果

Copy link
Collaborator

@zhangjun zhangjun left a comment

Choose a reason for hiding this comment

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

LGTM

@zhangjun zhangjun merged commit 65eb35f into PaddlePaddle:develop Nov 22, 2021
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.

2 participants