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

Tensor indexing #3195

Merged
merged 24 commits into from Aug 3, 2021
Merged

Tensor indexing #3195

merged 24 commits into from Aug 3, 2021

Commits on Jul 29, 2021

  1. CPU tensor indexing. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    218c1ba View commit details
    Browse the repository at this point in the history

  2. Add python wrapper for slicing. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    039fd7a View commit details
    Browse the repository at this point in the history

  3. * Add GPU subscript operator 

    * Add layout handling
* Add Python front-end for slicing

Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    48e3326 View commit details
    Browse the repository at this point in the history

  4. Reduce max. non-collapsible dims to 16. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    24e4c9c View commit details
    Browse the repository at this point in the history

  5. Fix output layout calculation. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    771da06 View commit details
    Browse the repository at this point in the history

  6. Enable tensor arguments. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    6269261 View commit details
    Browse the repository at this point in the history

  7. Add tests. Fix ExpandDims for 0D input and nonempty new_axis_names. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    887cf40 View commit details
    Browse the repository at this point in the history

  8. Add cpu-only and variable batch size tests. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    6f28e43 View commit details
    Browse the repository at this point in the history

  9. Fix behaviour with swapped ends. Fix a bug in TensorVector - type not… 

    … set on views when pointer or size is not changed.

Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    bb15eb9 View commit details
    Browse the repository at this point in the history

  10. Fix out-of-range error message. Add tests for out-of-range and incons… 

    …istent args.

Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    0ac474e View commit details
    Browse the repository at this point in the history

  11. Add missing file with tests. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    847f53e View commit details
    Browse the repository at this point in the history

  12. Fix review issues. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    f4146fa View commit details
    Browse the repository at this point in the history

  13. Prevent iteration over DataNode. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    1d8f677 View commit details
    Browse the repository at this point in the history

  14. Review issues. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 29, 2021
    Copy the full SHA
    86ba294 View commit details
    Browse the repository at this point in the history

Commits on Jul 30, 2021

  1. Add a hack to suppress DataNode being detected as an instance of Iter… 

    …able.

Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Jul 30, 2021
    Copy the full SHA
    601935e View commit details
    Browse the repository at this point in the history

Commits on Aug 2, 2021

  1. Add check for the number of indices. Fix some review issues. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 2, 2021
    Copy the full SHA
    d0e3369 View commit details
    Browse the repository at this point in the history

  2. Add more tests. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 2, 2021
    Copy the full SHA
    2391452 View commit details
    Browse the repository at this point in the history

  3. Add comments. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 2, 2021
    Copy the full SHA
    981923a View commit details
    Browse the repository at this point in the history

  4. Fix Python LGTM. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 2, 2021
    Copy the full SHA
    2726dd1 View commit details
    Browse the repository at this point in the history

Commits on Aug 3, 2021

  1. * Add message check to error tests. 

    * Fix some tests.
* Fix review issues.

Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 3, 2021
    Copy the full SHA
    f9eab5a View commit details
    Browse the repository at this point in the history

  2. Remove print. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 3, 2021
    Copy the full SHA
    6e878c8 View commit details
    Browse the repository at this point in the history

  3. Fix trailin full-range handling. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 3, 2021
    Copy the full SHA
    d93320a View commit details
    Browse the repository at this point in the history

  4. Add comments. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 3, 2021
    Copy the full SHA
    a6d6e43 View commit details
    Browse the repository at this point in the history

  5. Add subscript_dim_check to cpu_only and variable_batch_size tests. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Aug 3, 2021
    Copy the full SHA
    13961ae View commit details
    Browse the repository at this point in the history