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

feat: to/from PyTorch JaggedTensor #3246

Draft
wants to merge 20 commits into
base: main
Choose a base branch
from
Draft

feat: to/from PyTorch JaggedTensor #3246

wants to merge 20 commits into from
+372 −0

Commits on Sep 17, 2024

  1. add new to_jaggedtensor function

    @maxymnaumchyk
    maxymnaumchyk committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    9e28cb1 View commit details
    Browse the repository at this point in the history

  2. update __init__.py file

    @maxymnaumchyk
    maxymnaumchyk committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    14b09d2 View commit details
    Browse the repository at this point in the history

  3. add imports

    @maxymnaumchyk
    maxymnaumchyk committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    70a2630 View commit details
    Browse the repository at this point in the history

  4. style: pre-commit fixes

    @pre-commit-ci
    pre-commit-ci[bot] committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    9feb758 View commit details
    Browse the repository at this point in the history

  5. add style changes

    @maxymnaumchyk
    maxymnaumchyk committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    4674108 View commit details
    Browse the repository at this point in the history

  6. Merge remote-tracking branch 'origin/maxymnaumchyk/add-to-from-jagged… 

    …tensor-functions' into maxymnaumchyk/add-to-from-jaggedtensor-functions

# Conflicts:
#	src/awkward/operations/ak_to_jaggedtensor.py
    @maxymnaumchyk
    maxymnaumchyk committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    c0da418 View commit details
    Browse the repository at this point in the history

  7. add style changes

    @maxymnaumchyk
    maxymnaumchyk committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    000e10a View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2024

  1. check if fbgemm is installed

    @maxymnaumchyk
    maxymnaumchyk committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    b820b51 View commit details
    Browse the repository at this point in the history

  2. correct checking if fbgemm is installed

    @maxymnaumchyk
    maxymnaumchyk committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    ade482e View commit details
    Browse the repository at this point in the history

  3. add cuda support

    @maxymnaumchyk
    maxymnaumchyk committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    fc97f37 View commit details
    Browse the repository at this point in the history

  4. add comments

    @maxymnaumchyk
    maxymnaumchyk committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    0f48a9a View commit details
    Browse the repository at this point in the history

  5. style changes

    @maxymnaumchyk
    maxymnaumchyk committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    5e68ab5 View commit details
    Browse the repository at this point in the history

  6. fix tests

    @maxymnaumchyk
    maxymnaumchyk committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    2cdc0db View commit details
    Browse the repository at this point in the history

Commits on Sep 19, 2024

  1. leave out the "padded" and "keep_regular" arguments

    @maxymnaumchyk
    maxymnaumchyk committed Sep 19, 2024
    Configuration menu
    Copy the full SHA
    5d2eaad View commit details
    Browse the repository at this point in the history

  2. add error for non existing backend

    @maxymnaumchyk
    maxymnaumchyk committed Sep 19, 2024
    Configuration menu
    Copy the full SHA
    81fd78a View commit details
    Browse the repository at this point in the history

  3. *better* handle backend stated by user

    @maxymnaumchyk
    maxymnaumchyk committed Sep 19, 2024
    Configuration menu
    Copy the full SHA
    4244642 View commit details
    Browse the repository at this point in the history

Commits on Sep 20, 2024

  1. separately handle cupy arrays

    @maxymnaumchyk
    maxymnaumchyk committed Sep 20, 2024
    Configuration menu
    Copy the full SHA
    86d5922 View commit details
    Browse the repository at this point in the history

  2. delete "backend" argument since the same can be done with ak.to_backe… 

    …nd()
    @maxymnaumchyk
    maxymnaumchyk committed Sep 20, 2024
    Configuration menu
    Copy the full SHA
    b316181 View commit details
    Browse the repository at this point in the history

  3. add new "to_jaggedtensor" function

    @maxymnaumchyk
    maxymnaumchyk committed Sep 20, 2024
    Configuration menu
    Copy the full SHA
    d1d723b View commit details
    Browse the repository at this point in the history

  4. style fixes

    @maxymnaumchyk
    maxymnaumchyk committed Sep 20, 2024
    Configuration menu
    Copy the full SHA
    940b466 View commit details
    Browse the repository at this point in the history