Skip to content

Add decorator for extending a native generic function#6764

Closed
jtuyls wants to merge 4 commits intoapache:mainfrom
jtuyls:extend-native-generic-func
Closed

Add decorator for extending a native generic function#6764
jtuyls wants to merge 4 commits intoapache:mainfrom
jtuyls:extend-native-generic-func

Conversation

@jtuyls
Copy link
Contributor

@jtuyls jtuyls commented Oct 26, 2020

Decorator API for adding a new specialization of a native generic function for a specific target. This is for example useful for adding target specific strategies.

@tvm.target.extend_native_generic_func("add_strategy", "mytarget")
def add_strategy_mytarget(attrs, inputs, out_type, target):
    # Adds a specialization of the addition strategy for 'mytarget'
    ...

@tqchen
Copy link
Member

tqchen commented Apr 26, 2021

close due to inactive status, feel free to reopen a new pr

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants