Searched defs:smem (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/test/Examples/NVGPU/tools/ |
| H A D | nvdsl.py | 204 smem=None, argument
235 def update_smem(self, smem): argument
308 def mlir_gpu_launch(grid=(1, 1, 1), block=(1, 1, 1), smem=0): argument
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | RocmRuntimeWrappers.cpp | 63 intptr_t blockZ, int32_t smem, in mgpuLaunchKernel() argument
|
| H A D | CudaRuntimeWrappers.cpp | 169 intptr_t blockZ, int32_t smem, CUstream stream, void **params, in mgpuLaunchKernel() argument
341 mgpuLaunchClusterKernel(CUfunction function,intptr_t clusterX,intptr_t clusterY,intptr_t clusterZ,intptr_t gridX,intptr_t gridY,intptr_t gridZ,intptr_t blockX,intptr_t blockY,intptr_t blockZ,int32_t smem,CUstream stream,void ** params,void ** extra,size_t) mgpuLaunchClusterKernel() argument
[all...] |