Skip to content
This repository has been archived by the owner on Dec 16, 2022. It is now read-only.

Data loading cuda device #4879

Merged
merged 18 commits into from
Jan 11, 2021
Merged

Data loading cuda device #4879

merged 18 commits into from
Jan 11, 2021

Commits on Dec 21, 2020

  1. add test with tensor fields

    @epwalsh
    epwalsh committed Dec 21, 2020
    Configuration menu
    Copy the full SHA
    67a332c View commit details
    Browse the repository at this point in the history

  2. improve nn.util.move_to_device

    @epwalsh
    epwalsh committed Dec 21, 2020
    Configuration menu
    Copy the full SHA
    dfb4de9 View commit details
    Browse the repository at this point in the history

  3. ensure start_method is 'spawn' when using lazy and mem pin

    @epwalsh
    epwalsh committed Dec 21, 2020
    Configuration menu
    Copy the full SHA
    bc42562 View commit details
    Browse the repository at this point in the history

  4. add 'non_blocking' arg to 'move_to_device'

    @epwalsh
    epwalsh committed Dec 21, 2020
    Configuration menu
    Copy the full SHA
    6f745b6 View commit details
    Browse the repository at this point in the history

  5. fix fake test tensor

    @epwalsh
    epwalsh committed Dec 21, 2020
    Configuration menu
    Copy the full SHA
    258565f View commit details
    Browse the repository at this point in the history

  6. fix sampler test

    @epwalsh
    epwalsh committed Dec 21, 2020
    Configuration menu
    Copy the full SHA
    fa185eb View commit details
    Browse the repository at this point in the history

  7. lint

    @epwalsh
    epwalsh committed Dec 21, 2020
    Configuration menu
    Copy the full SHA
    c39b6f8 View commit details
    Browse the repository at this point in the history

  8. fix 'move_to_device'

    @epwalsh
    epwalsh committed Dec 21, 2020
    Configuration menu
    Copy the full SHA
    c58e793 View commit details
    Browse the repository at this point in the history

  9. Merge branch 'vision' into data-loader-device

    @epwalsh
    epwalsh authored Dec 21, 2020
    Configuration menu
    Copy the full SHA
    1b4fe02 View commit details
    Browse the repository at this point in the history

Commits on Dec 22, 2020

  1. fix condition check

    @epwalsh
    epwalsh committed Dec 22, 2020
    Configuration menu
    Copy the full SHA
    7416895 View commit details
    Browse the repository at this point in the history

Commits on Jan 5, 2021

  1. add device to data loader

    @epwalsh
    epwalsh committed Jan 5, 2021
    Configuration menu
    Copy the full SHA
    7ad0bd4 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'vision' into data-loader-device

    @epwalsh
    epwalsh authored Jan 5, 2021
    Configuration menu
    Copy the full SHA
    6fcd785 View commit details
    Browse the repository at this point in the history

  3. clean up doc string

    @epwalsh
    epwalsh committed Jan 5, 2021
    Configuration menu
    Copy the full SHA
    7d61a51 View commit details
    Browse the repository at this point in the history

  4. rename 'device' arg to 'cuda_device'

    @epwalsh
    epwalsh committed Jan 5, 2021
    Configuration menu
    Copy the full SHA
    d2d6929 View commit details
    Browse the repository at this point in the history

Commits on Jan 8, 2021

  1. pinning is very slow, revert

    @epwalsh
    epwalsh committed Jan 8, 2021
    Configuration menu
    Copy the full SHA
    086f109 View commit details
    Browse the repository at this point in the history

  2. DataLoaders load to CUDA device

    @epwalsh
    epwalsh committed Jan 8, 2021
    Configuration menu
    Copy the full SHA
    546e6f7 View commit details
    Browse the repository at this point in the history

  3. fix evaluate test

    @epwalsh
    epwalsh committed Jan 8, 2021
    Configuration menu
    Copy the full SHA
    09a0efa View commit details
    Browse the repository at this point in the history

Commits on Jan 11, 2021

  1. Merge branch 'vision' into data-loader-device

    @epwalsh
    epwalsh authored Jan 11, 2021
    Configuration menu
    Copy the full SHA
    b7fa447 View commit details
    Browse the repository at this point in the history