Lines Matching defs:CUstream

169                  intptr_t blockZ, int32_t smem, CUstream stream, void **params,
198 extern "C" MLIR_CUDA_WRAPPERS_EXPORT CUstream mgpuStreamCreate() {
200   CUstream stream = nullptr;
205 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuStreamDestroy(CUstream stream) {
210 mgpuStreamSynchronize(CUstream stream) {
214 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuStreamWaitEvent(CUstream stream,
235                                                           CUstream stream) {
240 mgpuMemAlloc(uint64_t sizeBytes, CUstream stream, bool isHostShared) {
256                                                       CUstream /*stream*/) {
261 mgpuMemcpy(void *dst, void *src, size_t sizeBytes, CUstream stream) {
268 mgpuMemset32(void *dst, unsigned int value, size_t count, CUstream stream) {
274 mgpuMemset16(void *dst, unsigned short value, size_t count, CUstream stream) {
349     CUstream stream, void **params, void **extra, size_t /*paramsCount*/) {
567 mgpuCreateDnVec(intptr_t size, void *values, int32_t dtp, CUstream /*stream*/) {
575 mgpuDestroyDnVec(void *v, CUstream /*stream*/) {
582                 CUstream /*stream*/) {
591 mgpuDestroyDnMat(void *m, CUstream /*stream*/) {
599               CUstream /*stream*/) {
612                  void *values, int32_t itp, int32_t dtp, CUstream /*stream*/) {
625               int32_t dtp, CUstream /*stream*/) {
639               int32_t dtp, CUstream /*stream*/) {
653               int32_t ptp, int32_t itp, int32_t dtp, CUstream /*stream*/) {
667 mgpuDestroySpMat(void *m, CUstream /*stream*/) {
673     int32_t ma, void *a, void *x, void *y, int32_t ctp, CUstream /*stream*/) {
691                                                    CUstream /*stream*/) {
706                    int32_t ctp, CUstream /*stream*/) {
725                                                    CUstream /*stream*/) {
741                     int32_t ctp, CUstream /*stream*/) {
760                                                     CUstream /*stream*/) {
775 mgpuSpGEMMCreateDescr(CUstream /*stream*/) {
782 mgpuSpGEMMDestroyDescr(void *s, CUstream /*stream*/) {
789     intptr_t bs, void *buf, CUstream /*stream*/) {
807                   int32_t ctp, intptr_t bsz2, void *buf2, CUstream /*stream*/) {
825                int32_t ctp, CUstream /*stream*/) {
840 mgpuSpMatGetSize(void *m, void *r, void *c, void *n, CUstream /*stream*/) {
850 mgpuSetCsrPointers(void *m, void *p, void *c, void *v, CUstream /*stream*/) {
902                           int32_t dtp, CUstream /*stream*/) {
915 mgpuDestroyCuSparseLtDnMat(void *dh, CUstream /*stream*/) {
922                               void *values, int32_t dtp, CUstream /*stream*/) {
935 mgpuDestroyCuSparseLtSpMat(void *sh, CUstream /*stream*/) {
948                              CUstream stream) {
1007                    CUstream stream) {