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

Multinomial #141

Merged
merged 56 commits into from
Sep 2, 2024
Merged

Multinomial #141

merged 56 commits into from
Sep 2, 2024

Commits on Jul 28, 2024

  1. WIP: multinomial

    @tongxin
    tongxin committed Jul 28, 2024
    Configuration menu
    Copy the full SHA
    1575860 View commit details
    Browse the repository at this point in the history

Commits on Jul 29, 2024

  1. add Ops & UT & Bench

    Bowen12992 committed Jul 29, 2024
    Configuration menu
    Copy the full SHA
    4e4879e View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' of https://github.com/FlagOpen/FlagGems into op… 

    …_dev
    Bowen12992 committed Jul 29, 2024
    Configuration menu
    Copy the full SHA
    00da77e View commit details
    Browse the repository at this point in the history

Commits on Jul 30, 2024

  1. add full zero ones Ops & UT & Bench

    Bowen12992 committed Jul 30, 2024
    Configuration menu
    Copy the full SHA
    22e14de View commit details
    Browse the repository at this point in the history

  2. split normal op

    Bowen12992 committed Jul 30, 2024
    Configuration menu
    Copy the full SHA
    5063551 View commit details
    Browse the repository at this point in the history

  3. Adding multinomial.

    @tongxin
    tongxin committed Jul 30, 2024
    Configuration menu
    Copy the full SHA
    55987a6 View commit details
    Browse the repository at this point in the history

Commits on Jul 31, 2024

  1. fixed one off error in binary search

    @tongxin
    tongxin committed Jul 31, 2024
    Configuration menu
    Copy the full SHA
    0f6822a View commit details
    Browse the repository at this point in the history

Commits on Aug 1, 2024

  1. Added multinomial tests without replacement.

    @tongxin
    tongxin committed Aug 1, 2024
    Configuration menu
    Copy the full SHA
    c894eb3 View commit details
    Browse the repository at this point in the history

  2. PR comment

    Bowen12992 committed Aug 1, 2024
    Configuration menu
    Copy the full SHA
    521dd05 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'master' of https://github.com/FlagOpen/FlagGems into op… 

    …_dev
    Bowen12992 committed Aug 1, 2024
    Configuration menu
    Copy the full SHA
    f39ae12 View commit details
    Browse the repository at this point in the history

  4. split test_special_ops

    Bowen12992 committed Aug 1, 2024
    Configuration menu
    Copy the full SHA
    732edca View commit details
    Browse the repository at this point in the history

  5. updated with_replacement tests

    @tongxin
    tongxin committed Aug 1, 2024
    Configuration menu
    Copy the full SHA
    85e93b3 View commit details
    Browse the repository at this point in the history

  6. add K-S test

    Bowen12992 committed Aug 1, 2024
    Configuration menu
    Copy the full SHA
    145ed76 View commit details
    Browse the repository at this point in the history

  7. split special perf

    Bowen12992 committed Aug 1, 2024
    Configuration menu
    Copy the full SHA
    273efc8 View commit details
    Browse the repository at this point in the history

  8. Update to a more reliable without-replacement test

    @tongxin
    tongxin committed Aug 1, 2024
    Configuration menu
    Copy the full SHA
    f46da90 View commit details
    Browse the repository at this point in the history

  9. Exponential added. (#138) 

    * exponential added.
* Added K-S tests to exponential_, fp64 corrected.
* aligned with aten prototype
* Exponential_ uses uint64 offsets in Triton kernel.
* Update pyproject config for new test dependencies.
    @tongxin
    tongxin authored Aug 1, 2024
    Configuration menu
    Copy the full SHA
    4a5acdd View commit details
    Browse the repository at this point in the history

Commits on Aug 2, 2024

  1. table

    Bowen12992 committed Aug 2, 2024
    Configuration menu
    Copy the full SHA
    2f725c4 View commit details
    Browse the repository at this point in the history

  2. resolve conflict

    Bowen12992 committed Aug 2, 2024
    Configuration menu
    Copy the full SHA
    f1b27d1 View commit details
    Browse the repository at this point in the history

  3. Use int64 indexing when needed & fix argmax (#146) 

     1. fix amax, armax and triu, use int64 indexing when the largest tensor's size_in_bytes exceed int32's max;
2. change the tiling scheme for argmax to loop in the reduction dimension, instead of data-size-dependent-tile-size
    @iclementine
    iclementine authored Aug 2, 2024
    Configuration menu
    Copy the full SHA
    9e878c4 View commit details
    Browse the repository at this point in the history

  4. test for op

    Bowen12992 committed Aug 2, 2024
    Configuration menu
    Copy the full SHA
    371f108 View commit details
    Browse the repository at this point in the history

  5. test for op

    Bowen12992 committed Aug 2, 2024
    Configuration menu
    Copy the full SHA
    6fcc8e7 View commit details
    Browse the repository at this point in the history

Commits on Aug 4, 2024

  1. Added multinomial perf tests.

    @tongxin
    tongxin committed Aug 4, 2024
    Configuration menu
    Copy the full SHA
    a7feb4b View commit details
    Browse the repository at this point in the history

Commits on Aug 5, 2024

  1. Making libentry thread safe (#136) 

    * libentry now is lock protected.

* Add multithreading tests for libentry.

* polish code.
    @tongxin
    tongxin authored Aug 5, 2024
    Configuration menu
    Copy the full SHA
    369aa82 View commit details
    Browse the repository at this point in the history

Commits on Aug 6, 2024

  1. add argparse

    Bowen12992 committed Aug 6, 2024
    Configuration menu
    Copy the full SHA
    423731d View commit details
    Browse the repository at this point in the history

  2. fix desc

    Bowen12992 committed Aug 6, 2024
    Configuration menu
    Copy the full SHA
    0da6086 View commit details
    Browse the repository at this point in the history

  3. fix num

    Bowen12992 committed Aug 6, 2024
    Configuration menu
    Copy the full SHA
    de4b098 View commit details
    Browse the repository at this point in the history

  4. Update test_specific_ops.py

    @Bowen12992
    Bowen12992 authored Aug 6, 2024
    Configuration menu
    Copy the full SHA
    980c9d7 View commit details
    Browse the repository at this point in the history

  5. Merge pull request #151 from Bowen12992/test_for_op 

    [Test] Test for op
    @Bowen12992
    Bowen12992 authored Aug 6, 2024
    Configuration menu
    Copy the full SHA
    e77db4c View commit details
    Browse the repository at this point in the history

  6. split UT files

    Bowen12992 committed Aug 6, 2024
    Configuration menu
    Copy the full SHA
    22654f1 View commit details
    Browse the repository at this point in the history

  7. Merge branch 'master' of https://github.com/FlagOpen/FlagGems into op… 

    …_dev
    Bowen12992 committed Aug 6, 2024
    Configuration menu
    Copy the full SHA
    78bdaf2 View commit details
    Browse the repository at this point in the history

  8. fix

    Bowen12992 committed Aug 6, 2024
    Configuration menu
    Copy the full SHA
    0dde1a6 View commit details
    Browse the repository at this point in the history

  9. fix

    Bowen12992 committed Aug 6, 2024
    Configuration menu
    Copy the full SHA
    34c7522 View commit details
    Browse the repository at this point in the history

  10. Merge pull request #139 from Bowen12992/op_dev 

    [Operator] Add uniform, normal, resolve_neg & resolve_conj & zeros & ones & full Ops with UT & Bench
    @Bowen12992
    Bowen12992 authored Aug 6, 2024
    Configuration menu
    Copy the full SHA
    7042de1 View commit details
    Browse the repository at this point in the history

  11. Merge branch 'master' into multinomial

    @tongxin
    tongxin authored Aug 6, 2024
    Configuration menu
    Copy the full SHA
    dc9ce0c View commit details
    Browse the repository at this point in the history

  12. resolved conflicts with master.

    @tongxin
    tongxin committed Aug 6, 2024
    Configuration menu
    Copy the full SHA
    ce37522 View commit details
    Browse the repository at this point in the history

Commits on Aug 7, 2024

  1. Move multinomial hypothesis tests to test_distribution_ops, resolve c… 

    …onflicts with master.
    @tongxin
    tongxin committed Aug 7, 2024
    Configuration menu
    Copy the full SHA
    376344c View commit details
    Browse the repository at this point in the history

Commits on Aug 19, 2024

  1. fixing multinomial, working in progress.

    @tongxin
    tongxin committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    db5dd1e View commit details
    Browse the repository at this point in the history

  2. Multinomial passes tests.

    @tongxin
    tongxin committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    6c6a612 View commit details
    Browse the repository at this point in the history

  3. Enhance multinomial tests and benchmarks.

    @tongxin
    tongxin committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    3d40f7a View commit details
    Browse the repository at this point in the history

  4. resolve conflicts.

    @tongxin
    tongxin committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    134bef1 View commit details
    Browse the repository at this point in the history

Commits on Aug 20, 2024

  1. Merge branch 'master' into multinomial

    @StrongSpoon
    StrongSpoon authored Aug 20, 2024
    Configuration menu
    Copy the full SHA
    e25e04e View commit details
    Browse the repository at this point in the history

  2. [bugfix] keepdim when samples one

    @StrongSpoon
    StrongSpoon committed Aug 20, 2024
    Configuration menu
    Copy the full SHA
    e7b2c5d View commit details
    Browse the repository at this point in the history

Commits on Aug 21, 2024

  1. [bugfix] fix accu test

    @StrongSpoon
    StrongSpoon committed Aug 21, 2024
    Configuration menu
    Copy the full SHA
    47e76ad View commit details
    Browse the repository at this point in the history

Commits on Aug 25, 2024

  1. fix anomaly behavior in fused_renorm_cumsum

    @tongxin
    tongxin committed Aug 25, 2024
    Configuration menu
    Copy the full SHA
    55c84f7 View commit details
    Browse the repository at this point in the history

Commits on Aug 26, 2024

  1. Polish multinomial tests.

    @tongxin
    tongxin committed Aug 26, 2024
    Configuration menu
    Copy the full SHA
    aa66384 View commit details
    Browse the repository at this point in the history

  2. remove garbage files.

    @tongxin
    tongxin committed Aug 26, 2024
    Configuration menu
    Copy the full SHA
    650981d View commit details
    Browse the repository at this point in the history

  3. multinomial result casted to int64 for n_samples==1

    @tongxin
    tongxin committed Aug 26, 2024
    Configuration menu
    Copy the full SHA
    71f1860 View commit details
    Browse the repository at this point in the history

  4. Merge with master

    @tongxin
    tongxin committed Aug 26, 2024
    Configuration menu
    Copy the full SHA
    82e3a4d View commit details
    Browse the repository at this point in the history

Commits on Aug 28, 2024

  1. bfloat16 added for multinomial, polish without replacement test.

    @tongxin
    tongxin committed Aug 28, 2024
    Configuration menu
    Copy the full SHA
    a8f9a47 View commit details
    Browse the repository at this point in the history

  2. merged with master.

    @tongxin
    tongxin committed Aug 28, 2024
    Configuration menu
    Copy the full SHA
    3f6ec74 View commit details
    Browse the repository at this point in the history

Commits on Aug 31, 2024

  1. Enable two-pass normed cumsum.

    @tongxin
    tongxin committed Aug 31, 2024
    Configuration menu
    Copy the full SHA
    afd2c0d View commit details
    Browse the repository at this point in the history

  2. cumsum updated

    @tongxin
    tongxin committed Aug 31, 2024
    Configuration menu
    Copy the full SHA
    bfd62e5 View commit details
    Browse the repository at this point in the history

Commits on Sep 1, 2024

  1. normed cumsum complete.

    @tongxin
    tongxin committed Sep 1, 2024
    Configuration menu
    Copy the full SHA
    572853c View commit details
    Browse the repository at this point in the history

  2. Fixed multinomial binary search boundary bug

    @tongxin
    tongxin committed Sep 1, 2024
    Configuration menu
    Copy the full SHA
    85a472e View commit details
    Browse the repository at this point in the history

  3. fix normed_cumsum bugs.

    @tongxin
    tongxin committed Sep 1, 2024
    Configuration menu
    Copy the full SHA
    eafe33d View commit details
    Browse the repository at this point in the history

  4. quick fix dim check.

    @tongxin
    tongxin committed Sep 1, 2024
    Configuration menu
    Copy the full SHA
    459bb61 View commit details
    Browse the repository at this point in the history