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

Support MTIA device type in FBGEEM TBE training #1994

Closed
wants to merge 1 commit into from
Closed

Support MTIA device type in FBGEEM TBE training #1994

wants to merge 1 commit into from

Commits on Sep 6, 2023

  1. Support MTIA device type in FBGEEM TBE training (pytorch#1994) 

    Summary:

For the MTIA device type, we have to rely on the CPU fallback. So we let the FBGEMM chose CPU path when the device type == MTIA.

Reviewed By: jackm321

Differential Revision: D48809630
    @egienvalue @facebook-github-bot
    egienvalue authored and facebook-github-bot committed Sep 6, 2023
    Configuration menu
    Copy the full SHA
    73b1a38 View commit details
    Browse the repository at this point in the history