[all-commits] [llvm/llvm-project] 538814: [mlir][cuda runtime] Set Max Dynamic Shared Memory...

[Guray Ozen via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 53881490c2edae7cdaa186f0c371e4ef27e0ed5e
      https://github.com/llvm/llvm-project/commit/53881490c2edae7cdaa186f0c371e4ef27e0ed5e
  Author: Guray Ozen <guray.ozen at gmail.com>
  Date:   2023-08-02 (Wed, 02 Aug 2023)

  Changed paths:
    M mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp

  Log Message:
  -----------
  [mlir][cuda runtime] Set Max Dynamic Shared Memory Attribute

This works aims to address the issue related to larger shared memory usage in the MLIR CUDA runtime. Currently, when the shared memory usage exceeds 48KB, we need to set the CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES attribute of the CUDA kernel appropriately. This work takes care of that by setting the attribute as required. Additionally, it includes some debug prints for better visibility and troubleshooting.

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D156874