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

[CUDA] Support multiple TIR-level dynamic shared memory allocations #8571

Merged
merged 21 commits into from
Jul 31, 2021
Merged

[CUDA] Support multiple TIR-level dynamic shared memory allocations #8571

merged 21 commits into from
Jul 31, 2021

Commits on Jul 27, 2021

  1. add shared mem matmul test

    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    617d7e0 View commit details
    Browse the repository at this point in the history

  2. Add a stub pass

    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    26bfb17 View commit details
    Browse the repository at this point in the history

  3. add builtin for reinterprete load/store

    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    9eab8d3 View commit details
    Browse the repository at this point in the history

  4. remove buitlin since Load/Store node already support reinterpret

    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    27c881a View commit details
    Browse the repository at this point in the history

  5. Add Load/Store visitor implementation

    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    f1e35fa View commit details
    Browse the repository at this point in the history

  6. allocate merge first cut

    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    482ecd8 View commit details
    Browse the repository at this point in the history

  7. Remove all attr::storage_scope usage

    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    ce62d9e View commit details
    Browse the repository at this point in the history

  8. fix allocate location

    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    c95ede5 View commit details
    Browse the repository at this point in the history

  9. Revert "Remove all attr::storage_scope usage" 

    This reverts commit ce62d9e.
    masa committed Jul 27, 2021
    Configuration menu
    Copy the full SHA
    d55ce65 View commit details
    Browse the repository at this point in the history

Commits on Jul 28, 2021

  1. handle vector alloc

    masa committed Jul 28, 2021
    Configuration menu
    Copy the full SHA
    c2892b6 View commit details
    Browse the repository at this point in the history

  2. add vectorized test

    masa committed Jul 28, 2021
    Configuration menu
    Copy the full SHA
    e8907f1 View commit details
    Browse the repository at this point in the history

  3. drop multi-lane dtype allocation support, vectorized store working

    masa committed Jul 28, 2021
    Configuration menu
    Copy the full SHA
    1f5c79c View commit details
    Browse the repository at this point in the history

  4. doc update

    masa committed Jul 28, 2021
    Configuration menu
    Copy the full SHA
    6d6f437 View commit details
    Browse the repository at this point in the history

  5. dtype fix in the test

    masa committed Jul 28, 2021
    Configuration menu
    Copy the full SHA
    f086e37 View commit details
    Browse the repository at this point in the history

  6. lint fix, do not run merging when number of alloc is 1

    masa committed Jul 28, 2021
    Configuration menu
    Copy the full SHA
    391b2c6 View commit details
    Browse the repository at this point in the history

Commits on Jul 29, 2021

  1. Update src/tir/transforms/merge_dynamic_shared_memory_allocations.cc 

    Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
    @masahi @Hzfengsy
    masahi and Hzfengsy authored Jul 29, 2021
    Configuration menu
    Copy the full SHA
    6203b0b View commit details
    Browse the repository at this point in the history

  2. Update src/tir/transforms/merge_dynamic_shared_memory_allocations.cc 

    Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
    @masahi @Hzfengsy
    masahi and Hzfengsy authored Jul 29, 2021
    Configuration menu
    Copy the full SHA
    56f8dce View commit details
    Browse the repository at this point in the history

  3. move test cases into dedicated file

    masa committed Jul 29, 2021
    Configuration menu
    Copy the full SHA
    ec3d0da View commit details
    Browse the repository at this point in the history

  4. use integer division

    masa committed Jul 29, 2021
    Configuration menu
    Copy the full SHA
    30bbbbb View commit details
    Browse the repository at this point in the history

  5. verify new transform pass output

    masa committed Jul 29, 2021
    Configuration menu
    Copy the full SHA
    b46e1f2 View commit details
    Browse the repository at this point in the history

Commits on Jul 30, 2021

  1. add test on combined buffer reuse and merge

    @masahi
    masahi committed Jul 30, 2021
    Configuration menu
    Copy the full SHA
    7a37406 View commit details
    Browse the repository at this point in the history