[dataflow][backend] Add IO analysis pass

[chhzh123]

Description

This PR fixes the issue of incorrect dataflow construction introduced by #263. Specifically, we need to analyze the arguments in the top-level function and check whether they have read/write operations. This is done by a sequential traversal of the IR through the FuncOp.

Examples

def kernel(A: int32[32], B: int32[32], C: int32[32]):
    for i in range(32):
        C[i] = A[i] + B[i]

def rw_kernel(D: int32[32]):
    for i in range(32):
        D[i] = D[i] + 1

def top2(A: int32[32], B: int32[32], C: int32[32], D: int32[32]):
    kernel(A, B, C)
    write_kernel(A)
    rw_kernel(D)

s = allo.customize(top2)
res = analyze_arg_load_store(s.module)

{'kernel': ['in', 'in', 'out'], 'write_kernel': ['out'], 'rw_kernel': ['both'], 'top2': ['both', 'in', 'out', 'both']}

Checklist

• PR's title starts with a category (e.g. [Bugfix], [IR], [Builder], etc)
• Changes are complete (i.e. I finished coding on this PR)
• All changes have test coverage (It would be better to provide ~2 different test cases to test the robustness of your code)
• Code is well-documented