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

[TIR] Do not drop 4th argument to tir.max #15763

Merged

Commits on Sep 15, 2023

  1. [TIR] Do not drop 4th argument to tir.max

    The `tir.op.comm_reducer` utility provides two distinct APIs, either
    reducing along a tensor axis or reducing along a list of arguments.
    
    Prior to this commit, when reducing along a list of arguments, the 4th
    argument was silently dropped.  For example,
    `tvm.tir.max(1,2,3,4,3,2,1)` would return `3`.
    Lunderberg committed Sep 15, 2023
    Configuration menu
    Copy the full SHA
    9b0df1e View commit details
    Browse the repository at this point in the history