Gabriel Kosmacher

TeamPolicy gpu parallel for hangs on more that 5 serial calls.

5

I am running into an issue that when I am calling a pk.parallel_for loop sequentially on a gpu more than 5 time in a row, the code hangs. Here is...

PyTorch Support

1

When I try to pass a PyTorch tensor into a workunit as I would a CuPy array, as in the following script, ```python import torch import pykokkos as pk @pk.workunit...

In a `pk.workunit`, my code fails when I try to access a cupy array as `x[a, b]` instead of `x[a][b]`. For example, the script ```python import cupy as cp import...

Kokkos supports calling multiple levels of scratch memory as in the code below ```cpp policy.set_scratch_size(0, PerTeam(bytes0)) .set_scratch_size(1, PerTeam(bytes1)); ``` but pykokkos throws an error when I call ```python pk.TeamPolicy(league_size, team_size).set_scratch_size(0,...