Is it possible to call openblas library in TIR. Like below example

```
@tvm.script.ir_module
class MatmulModule:
    @T.prim_func
    def matmul(
        A: T.Buffer((128, 128), "float32"),
        B: T.Buffer((128, 128), "float32"),
        C: T.Buffer((128, 128), "float32"),
    ):
        T.func_attr({"tir.noalias": True})
        # for i, j, k in T.grid(128, 128, 128):
        #     with T.block("matmul"):
        #         vi, vj, vk = T.axis.remap("SSR", [i, j, k])
        #         with T.init():
        #             C[vi, vj] = 0.0
        #         C[vi, vj] += A[vi, vk] * B[vk, vj]
        C = tvm.tir.call_packed(cblas.matmul(A, B))
```





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/call-openblas-library-in-tir/18373/1) 
to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/bda99e995cc2db288459d6ff5f90c173c22bdf51019350256b57629297f4321b).

Reply via email to