aashaka 0650371b54 Allow obtaining cuda stream handle from PyTorch stream when launching kernel (#297) ...

Use `cuda_stream` attribute of a torch stream if the stream is not an
instance of the cupy stream.

2024-05-04 04:57:07 +00:00

..

__init__.py

Add executor to execute schedule-plan file (#283)

2024-04-18 19:10:41 +00:00

CMakeLists.txt

Fix bootstrapping mechanism (#278)

2024-03-27 10:24:24 +08:00

comm.py

Refactoring NVLS interfaces (#293)

2024-04-24 10:05:41 -07:00

core_py.cpp

Refactoring NVLS interfaces (#293)

2024-04-24 10:05:41 -07:00

error_py.cpp

Add executor to execute schedule-plan file (#283)

2024-04-18 19:10:41 +00:00

executor.cpp

Add executor to execute schedule-plan file (#283)

2024-04-18 19:10:41 +00:00

fifo_py.cpp

Update interface to let user change fifo size (#243)

2024-01-09 22:14:36 -08:00

numa_py.cpp

Pytest (#162)

2023-09-01 21:22:11 +08:00

nvls_py.cpp

Refactoring NVLS interfaces (#293)

2024-04-24 10:05:41 -07:00

proxy_channel_py.cpp

Update interface to let user change fifo size (#243)

2024-01-09 22:14:36 -08:00

semaphore_py.cpp

Release GIL for Python APIs with wait (#190)

2023-11-14 21:11:01 +08:00

sm_channel_py.cpp

Pytest (#162)

2023-09-01 21:22:11 +08:00

utils_py.cpp

NVLS support. (#250)

2024-02-04 20:46:10 -08:00

utils.py

Allow obtaining cuda stream handle from PyTorch stream when launching kernel (#297)

2024-05-04 04:57:07 +00:00