Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

rutkoor
Copy link
Contributor

@rutkoor rutkoor commented May 10, 2024

This PR introduces vector codegen support for llvm_pure_intrin builtin.

@rutkoor rutkoor force-pushed the vector_codegen_llvm_pure_intrin branch from deefe06 to d3be9ab Compare May 10, 2024 10:49
@rutkoor
Copy link
Contributor Author

rutkoor commented May 11, 2024

@tvm-bot rerun

@rutkoor
Copy link
Contributor Author

rutkoor commented May 14, 2024

cc: @tqchen @Lunderberg

@rutkoor
Copy link
Contributor Author

rutkoor commented May 16, 2024

cc: @cbalint13 @ekalda

@@ -139,7 +139,8 @@ TIR_DEFINE_BUILTIN_FUNC(call_llvm_intrin)
TIR_DEFINE_BUILTIN_FUNC(call_llvm_pure_intrin)
.set_attr<TCallEffectKind>("TCallEffectKind", Integer(CallEffectKind::kPure))
.set_attr<TScriptDtypePrintLocation>("TScriptDtypePrintLocation",
Integer(ScriptDtypePrintLocation::kFirst));
Integer(ScriptDtypePrintLocation::kFirst))
.set_attr<TVectorizable>("TVectorizable", true);
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

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

@cbalint13
Copy link
Contributor

cbalint13 commented May 16, 2024

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants