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

remove unused tensors from VK model's graph #4427

Closed
wants to merge 1 commit into from
Closed

remove unused tensors from VK model's graph #4427

wants to merge 1 commit into from

Commits on Jul 26, 2024

  1. remove unused tensors from VK model's graph (pytorch#4427) 

    Summary:
Pull Request resolved: pytorch#4427

We implemented [operators fusion](pytorch#3769) (`conv+bn`) which fused `conv` and `bn`'s weights and biases, but the old parameters are not deleted. Hence we saw that VK model's size is nearly twice large as CPU's.

As regards mobilenet_v2, before this diff CPU vs VK: 14M vs 22M. After this diff, both of them have 14M.

Reviewed By: SS-JIA

Differential Revision: D60257047
    @copyrightly @facebook-github-bot
    copyrightly authored and facebook-github-bot committed Jul 26, 2024
    Configuration menu
    Copy the full SHA
    c09fe87 View commit details
    Browse the repository at this point in the history