[CODEGEN] Vector-Codegen support for llvm-pure-intrin

[rutkoor]

If we have a loop with llvm intrinsic call like below,

Before,

for i in range T.serial(32):
    A[i] = T.call_llvm_pure_intrin(_ , _ , _ , B[i])

using sch.vectorize, we cannot generate ramp node for the above code.

In this pull request, I enabled vectorization support for the llvm_pure_intrin builtin. During the vectorize pass, I explicitly vectorized the compute expressions that serve as input to the intrinsic. As a result of these changes, I was able to generate the ramp instructions for the expressions inside the call_llvm_pure_intrin function.

After,

for i in range T.serial(...):
    A[T.Ramp( ... )] = T.call_llvm_pure_intrin(_ , _ , _ , B[T.Ramp( ... )])

Note: If the intrinsic doesn't support vector arguments as input, then the error will be thrown during llvm codegen.

[rutkoor]

@tvm-bot rerun

[rutkoor]

cc: @tqchen @Lunderberg

[rutkoor]

cc: @cbalint13 @ekalda

[cbalint13]

Can add a small testcase for the IR printer ?
tests/python/unittest/test_tvmscript_printer_tir.py

[cbalint13]

@rutkoor
Implementation LGTM (I did't go after the functionality).

[rutkoor]

@rutkoor Implementation LGTM (I did't go after the functionality).

Hi @cbalint13, thank you for your quick reply. And apologize for not replying back sooner.

I have added a testcase inside test_tvmscript_printer_tir.py and also explained what I am doing in the description of this PR. Please let me know if you want me to add more details.

Thanks.

[cbalint13]

@rutkoor Implementation LGTM (I did't go after the functionality).

Hi @cbalint13, thank you for your quick reply. And apologize for not replying back sooner.

I have added a testcase inside test_tvmscript_printer_tir.py and also explained what I am doing in the description of this PR. Please let me know if you want me to add more details.

Thank you very much @rutkoor !

• I had this issue [TVMScript][TIR] Pretty print TIR LLVM function name #15953 before, it is a good idea for such IR "expanders" to have a minimal "pretty print" test case, hence my request. Appologise to not giving more reference.

Thanks.

[rutkoor]

@tvm-bot rerun