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

[Inference/kernel]Add Fused Rotary Embedding and KVCache Memcopy CUDA Kernel #5418

Merged
merged 17 commits into from
Mar 13, 2024
Merged

[Inference/kernel]Add Fused Rotary Embedding and KVCache Memcopy CUDA Kernel #5418

merged 17 commits into from
Mar 13, 2024

Commits on Mar 1, 2024

  1. add rotary embedding kernel

    @isky-cd
    isky-cd committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    2d71702 View commit details
    Browse the repository at this point in the history

  2. add rotary_embedding_kernel

    @isky-cd
    isky-cd committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    e750303 View commit details
    Browse the repository at this point in the history

Commits on Mar 4, 2024

  1. add fused rotary_emb and kvcache memcopy

    @isky-cd
    isky-cd committed Mar 4, 2024
    Configuration menu
    Copy the full SHA
    94497b0 View commit details
    Browse the repository at this point in the history

  2. add fused_rotary_emb_and_cache_kernel.cu

    @isky-cd
    isky-cd committed Mar 4, 2024
    Configuration menu
    Copy the full SHA
    beb11e1 View commit details
    Browse the repository at this point in the history

  3. add fused_rotary_emb_and_memcopy

    @isky-cd
    isky-cd committed Mar 4, 2024
    Configuration menu
    Copy the full SHA
    8127ea0 View commit details
    Browse the repository at this point in the history

Commits on Mar 5, 2024

  1. fix bugs in fused_rotary_emb_and_cache_kernel.cu

    @isky-cd
    isky-cd committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    2abb2e1 View commit details
    Browse the repository at this point in the history

Commits on Mar 6, 2024

  1. fix ci bugs

    @isky-cd
    isky-cd committed Mar 6, 2024
    Configuration menu
    Copy the full SHA
    ac62d86 View commit details
    Browse the repository at this point in the history

Commits on Mar 7, 2024

  1. use vec memcopy and opt the gloabl memory access

    @isky-cd
    isky-cd committed Mar 7, 2024
    Configuration menu
    Copy the full SHA
    dfe8184 View commit details
    Browse the repository at this point in the history

Commits on Mar 8, 2024

  1. fix code style

    @isky-cd
    isky-cd committed Mar 8, 2024
    Configuration menu
    Copy the full SHA
    b1564c4 View commit details
    Browse the repository at this point in the history

  2. fix test_rotary_embdding_unpad.py

    @isky-cd
    isky-cd committed Mar 8, 2024
    Configuration menu
    Copy the full SHA
    2e024ac View commit details
    Browse the repository at this point in the history

  3. fix conflict

    @isky-cd
    isky-cd committed Mar 8, 2024
    Configuration menu
    Copy the full SHA
    1de7cc6 View commit details
    Browse the repository at this point in the history

  4. codes revised based on the review comments

    @isky-cd
    isky-cd committed Mar 8, 2024
    Configuration menu
    Copy the full SHA
    d2cedbf View commit details
    Browse the repository at this point in the history

Commits on Mar 13, 2024

  1. fix conflict

    @isky-cd
    isky-cd committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    0ea86d8 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'feature/colossal-infer' into rotary_embnedding_branch

    @isky-cd
    isky-cd committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    96cc7b0 View commit details
    Browse the repository at this point in the history

  3. fix bugs about include path

    @isky-cd
    isky-cd committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    629105c View commit details
    Browse the repository at this point in the history

  4. Merge branch 'feature/colossal-infer' into rotary_embnedding_branch

    @isky-cd
    isky-cd committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    95c183f View commit details
    Browse the repository at this point in the history

  5. rm inline

    @isky-cd
    isky-cd committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    6dfc68a View commit details
    Browse the repository at this point in the history