summaryrefslogtreecommitdiff
path: root/mlir/test/python/dialects/gpu
AgeCommit message (Collapse)Author
2025-11-18[MLIR][Python] Add arg_attrs and res_attrs to gpu func (#168475)Asher Mancinelli
I missed these attributes when I added the wrapper for GPUFuncOp in fbdd98f74f0d.
2025-10-20[mlir][python] Add Pythonic wrappers for gpu ops (#163883)Asher Mancinelli
Add builders on the Python side that match builders in the C++ side, add tests for launching GPU kernels and regions, and correct some small documentation mistakes. This reflects the API decisions already made in the func dialect's Python bindings and makes use of the GPU dialect's bindings work more similar to C++ interface.
2025-10-16[mlir][python] Add pythonic interface for GPUFuncOp (#163596)Asher Mancinelli
The func dialect provides a more pythonic interface for constructing operations, but the gpu dialect does not; this is the first PR to provide the same conveniences for the gpu dialect, starting with the gpu.func op.
2025-02-11[mlir][target][nvvm] Perf by stage and store into properties (#126178)Zichen Lu
Implement the feature about perf by stage(llvm-ir -> isa, isa->binary). The results will be stored into the properties, then users can use them after using GpuModuleToBinary Pass.
2024-08-27[mlir][gpu] Add metadata attributes for storing kernel metadata in GPU ↵Fabian Mora
objects (#95292) This patch adds the `#gpu.kernel_metadata` and `#gpu.kernel_table` attributes. The `#gpu.kernel_metadata` attribute allows storing metadata related to a compiled kernel, for example, the number of scalar registers used by the kernel. The attribute only has 2 required parameters, the name and function type. It also has 2 optional parameters, the arguments attributes and generic dictionary for storing all other metadata. The `#gpu.kernel_table` stores a table of `#gpu.kernel_metadata`, mapping the name of the kernel to the metadata. Finally, the function `ROCDL::getAMDHSAKernelsELFMetadata` was added to collect ELF metadata from a binary, and to test the class methods in both attributes. Example: ```mlir gpu.binary @binary [#gpu.object<#rocdl.target<chip = "gfx900">, kernels = #gpu.kernel_table<[ #gpu.kernel_metadata<"kernel0", (i32) -> (), metadata = {sgpr_count = 255}>, #gpu.kernel_metadata<"kernel1", (i32, f32) -> (), arg_attrs = [{llvm.read_only}, {}]> ]> , bin = "BLOB">] ``` The motivation behind these attributes is to provide useful information for things like tunning. --------- Co-authored-by: Mehdi Amini <joker.eph@gmail.com>
2024-04-18[mlir][python] add binding to `#gpu.object` (#88992)Maksim Levental
2024-02-29[mlir] Add result name for gpu.block_id and gpu.thread_id ops. (#83393)Alexander Belyaev
expand-arith-ops.mlir fails on windows, but this is unrelated to this PR
2023-11-04[MLIR][LLVM] Remove last typed pointer remnants from tests (#71232)Christian Ulmann
This commit removes all LLVM dialect typed pointers from the lit tests. Typed pointers have been deprecated for a while now and it's planned to soon remove them from the LLVM dialect. Related PSA: https://discourse.llvm.org/t/psa-removal-of-typed-pointers-from-the-llvm-dialect/74502
2023-10-30[mlir][python] Register LLVM translations in the RegisterEverything for ↵Jungwook Park
python (#70428) Added missing register_translations in python to replicate the same in the C-API Cleaned up the current calls to register passes where the other calls are already embedded in the mlirRegisterAllPasses. found here, https://discourse.llvm.org/t/opencl-example/74187