[Misc] Propagate signatures through expr decorators

[simondanielsson]

Motivation

Fixes #836.

This allows editors to show correct signatures of expressions rather than generic wrapper(*args, **kwargs) upon hover. This makes learning and using the DSL easier as it becomes more clear how to call expr functions.

Examples

1. Before:

With this PR: more clear how to call this function.

2. Before:

With this PR:

Technical Details

No runtime effect.

Test Plan

On MI300:

bash scripts/run_tests.sh    

Test Result

...
-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
======== 1325 passed, 2089 skipped, 934 deselected, 91 warnings in 687.01s (0:11:27) =========
acc=4
a=3 b=97
i=0
i=1
i=2
acc=4
[smem_probe] tid=0
[smem_probe] tid=0
[smem_probe] tid=0
[smem_probe] tid=0
[smem_probe] tid=0
[smem_probe] tid=0
[smem_probe] tid=0
[smem_probe] tid=0
[smem_probe] tid=0
kernel loop_count=1
helper i=0
kernel loop_count=4
helper i=0
helper i=1
helper i=2
helper i=3
kernel loop_count=8
helper i=0
helper i=1
helper i=2
helper i=3
helper i=4
helper i=5
helper i=6
helper i=7
[kernel] bid=0, tid=0
[kernel] bid=0, tid=1
[kernel] bid=0, tid=2

========================================================================
Examples (examples/)
========================================================================
  PASS  01-vectorAdd.py
  PASS  02-tiledCopy.py
  PASS  03-tiledMma.py
  PASS  04-preshuffle_gemm.py
  PASS  05-gather_scatter.py

========================================================================
MLIR FileCheck Tests
========================================================================
  PASS  Conversion/buffer_copy_lds.mlir
  PASS  Conversion/copy_atom_stateful.mlir
  PASS  Conversion/dyn_shared.mlir
  PASS  Conversion/gpu_ops.mlir
  PASS  Conversion/memref_ops.mlir
  PASS  Conversion/mma_atom.mlir
  PASS  Conversion/mma_atom_stateful.mlir
  PASS  Conversion/pointer_ops.mlir
  PASS  Conversion/type_conversion.mlir
  PASS  Conversion/wmma_gfx11.mlir
  PASS  Conversion/wmma_gfx11_neg.mlir
  PASS  LayoutAlgebra/coalesce.mlir
  PASS  LayoutAlgebra/composition.mlir
  PASS  LayoutAlgebra/construction.mlir
  PASS  LayoutAlgebra/coord_swizzle.mlir
  PASS  LayoutAlgebra/coordinate.mlir
  PASS  LayoutAlgebra/divide.mlir
  PASS  LayoutAlgebra/int_tuple.mlir
  PASS  LayoutAlgebra/mma_make_fragment_stages.mlir
  PASS  LayoutAlgebra/product.mlir
  PASS  LayoutAlgebra/size_cosize.mlir
  PASS  Transforms/canonicalize.mlir
  PASS  Transforms/convert-atom-call-to-ssa-form.mlir
  PASS  Transforms/expand_copy_reduced_pred.mlir
  PASS  Transforms/int_swizzle_simplify.mlir
  PASS  Transforms/layout_lowering.mlir
  PASS  Transforms/promote_regmem_to_ssa.mlir
  PASS  Transforms/promote_regmem_to_ssa_copy_pred.mlir
  PASS  Transforms/promote_regmem_to_ssa_invalid.mlir
  PASS  Transforms/rewrite_func_signature.mlir

========================================================================
All tests passed.
========================================================================

Submission Checklist

• Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

[simondanielsson]

@sjfeng1999 CI errors look unrelated, you could please take a look? Thanks