DALI icon indicating copy to clipboard operation
DALI copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA

Extend GPU numba support

Open JanuszL opened this issue 3 years ago • 6 comments

  • adds support for multiple inputs and outputs in Numba GPU functions
  • adds supports for non-uniform batches
  • fixes support inputs/outputs with more than 2 dimensions
  • extends GPU Numba tests based on the CPU one
  • aligns order of inputs and outputs in GPU Numba function with the CPU one

Signed-off-by: Janusz Lisiecki [email protected]

Category:

New feature (non-breaking change which adds functionality)

Description:

  • adds support for multiple inputs and outputs in Numba GPU functions
  • adds supports for non-uniform batches
  • fixes support inputs/outputs with more than 2 dimensions
  • extends GPU Numba tests based on the CPU one
  • aligns order of inputs and outputs in GPU Numba function with the CPU one

Additional information:

Affected modules and functionalities:

  • Numba operator for the GPU

Key points relevant for the review:

  • logic

Tests:

  • [ ] Existing tests apply
  • [x] New tests added
    • [x] Python tests
      • test_operator_numba_func.py
    • [ ] GTests
    • [ ] Benchmark
    • [ ] Other
  • [ ] N/A

Checklist

Documentation

  • [ ] Existing documentation applies
  • [x] Documentation updated
    • [x] Docstring
    • [ ] Doxygen
    • [ ] RST
    • [ ] Jupyter
    • [ ] Other
  • [ ] N/A

DALI team only

Requirements

  • [ ] Implements new requirements
  • [x] Affects existing requirements
  • [ ] N/A

REQ IDs: NUMBAF.02, NUMBAF.08

JIRA TASK: DALI-2856

JanuszL avatar Jun 23 '22 00:06 JanuszL

!build

JanuszL avatar Jun 23 '22 00:06 JanuszL

CI MESSAGE: [5153229]: BUILD STARTED

dali-automaton avatar Jun 23 '22 00:06 dali-automaton

CI MESSAGE: [5153229]: BUILD FAILED

dali-automaton avatar Jun 23 '22 02:06 dali-automaton

!build

JanuszL avatar Jun 23 '22 07:06 JanuszL

CI MESSAGE: [5156125]: BUILD STARTED

dali-automaton avatar Jun 23 '22 07:06 dali-automaton

CI MESSAGE: [5156125]: BUILD FAILED

dali-automaton avatar Jun 23 '22 08:06 dali-automaton

CI MESSAGE: [6418288]: BUILD FAILED

dali-automaton avatar Nov 07 '22 18:11 dali-automaton

CI MESSAGE: [8224582]: BUILD STARTED

dali-automaton avatar May 09 '23 06:05 dali-automaton

CI MESSAGE: [8224582]: BUILD FAILED

dali-automaton avatar May 09 '23 08:05 dali-automaton

!build

banasraf avatar Jul 11 '23 08:07 banasraf

CI MESSAGE: [8926955]: BUILD STARTED

dali-automaton avatar Jul 11 '23 08:07 dali-automaton

CI MESSAGE: [8926955]: BUILD PASSED

dali-automaton avatar Jul 11 '23 10:07 dali-automaton

CI MESSAGE: [8926955]: BUILD FAILED

dali-automaton avatar Jul 12 '23 09:07 dali-automaton

!build

banasraf avatar Jul 22 '23 18:07 banasraf

CI MESSAGE: [9072259]: BUILD STARTED

dali-automaton avatar Jul 22 '23 18:07 dali-automaton

CI MESSAGE: [9072259]: BUILD FAILED

dali-automaton avatar Jul 23 '23 01:07 dali-automaton

CI MESSAGE: [9072259]: BUILD PASSED

dali-automaton avatar Jul 24 '23 07:07 dali-automaton

!build

banasraf avatar Jul 28 '23 13:07 banasraf

CI MESSAGE: [9151455]: BUILD STARTED

dali-automaton avatar Jul 28 '23 13:07 dali-automaton

CI MESSAGE: [9151455]: BUILD PASSED

dali-automaton avatar Jul 28 '23 15:07 dali-automaton

!build

banasraf avatar Aug 24 '23 07:08 banasraf

CI MESSAGE: [9495343]: BUILD STARTED

dali-automaton avatar Aug 24 '23 07:08 dali-automaton

CI MESSAGE: [9495343]: BUILD PASSED

dali-automaton avatar Aug 24 '23 11:08 dali-automaton

!build

banasraf avatar Aug 25 '23 07:08 banasraf

CI MESSAGE: [9510722]: BUILD STARTED

dali-automaton avatar Aug 25 '23 07:08 dali-automaton

CI MESSAGE: [9510722]: BUILD PASSED

dali-automaton avatar Aug 25 '23 11:08 dali-automaton