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.

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

[Inductor][fx pass] Fuse pointwise operators in the post grad #114778

Closed
wants to merge 1 commit into from

Commits on Dec 7, 2023

  1. [Inductor][fx pass] Fuse pointwise operators in the post grad (pytorc…

    …h#114778)
    
    Summary:
    
    We construct a unified API that can be easily add pointwise ops to be batched in the post grad
    
    Test Plan:
    # unit test
    ```
    buck2 test 'fbcode//mode/dev-nosan' fbcode//caffe2/test/inductor:group_batch_fusion
    ```
    Buck UI: https://www.internalfb.com/buck2/6c5d1d31-e4d1-4865-bf79-1e7ac3b6e051
    Test UI: https://www.internalfb.com/intern/testinfra/testrun/1970325050015770
    Network: Up: 72KiB  Down: 22KiB  (reSessionID-44adc8b2-54e9-453a-bd20-710cefefaed1)
    Jobs completed: 20. Time elapsed: 1:44.6s.
    Cache hits: 0%. Commands: 2 (cached: 0, remote: 0, local: 2)
    Tests finished: Pass 7. Fail 0. Fatal 0. Skip 0. Build failure 0
    # local reproduce
    ### cmf
    P887605070
    ### igctr
    P892987433
    ### mai
    P893109069
    ### icvr
    P893075846
    ### oc
    P893109069
    ### mixed precison training
    P898569125
    
    # e2e test
    baseline
    f509792379
    proposal
    f509792025
    https://pxl.cl/3Xbcf
    
    Reviewed By: xuzhao9
    
    Differential Revision: D51332067
    mengluy authored and facebook-github-bot committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    1cda761 View commit details
    Browse the repository at this point in the history