| Name |
Last commit
|
History
|
Last Update |
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| common.h | ||
| cuda.h | ||
| dlpack.h | ||
| gradient.h | ||
| module.cc | ||
| mpi.h | ||
| numpy.h | ||
| plugin_op.cc | ||
| plugin_op.h | ||
| proto.h | ||
| sysconfig.h | ||
| tensor.h | ||
| types.h |
| Name |
Last commit
654febe3
-
Implement softmax kernels via warp reduce
|
History
|
Last Update |
|---|---|---|
| .. | ||
| CMakeLists.txt | Loading commit data... | |
| common.h | Loading commit data... | |
| cuda.h | Loading commit data... | |
| dlpack.h | Loading commit data... | |
| gradient.h | Loading commit data... | |
| module.cc | Loading commit data... | |
| mpi.h | Loading commit data... | |
| numpy.h | Loading commit data... | |
| plugin_op.cc | Loading commit data... | |
| plugin_op.h | Loading commit data... | |
| proto.h | Loading commit data... | |
| sysconfig.h | Loading commit data... | |
| tensor.h | Loading commit data... | |
| types.h | Loading commit data... |