We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
I do a test with scatterND:
def test_scatternd(): def before(): data = relay.const(np.zeros((1, 900, 300), dtype="float32"), dtype="float32") indices = relay.const(np.ones((3, 1, 900, 300), dtype="int64"), dtype="int64") update = relay.const(np.ones((1, 900, 300), dtype="float32"), dtype="float32") b = relay.op.scatter_nd(data, indices, update) return relay.Function(relay.analysis.free_vars(b), b) passes = tvm.transform.Sequential( [ relay.transform.InferType(), relay.transform.FoldConstant(), ] ) before_mod = tvm.IRModule.from_expr(before()) with tvm.transform.PassContext(opt_level=3): after_mod = passes(before_mod)
I found if the shape is (1, 900, 200), the after_mod will success execute. and if I use the shape of (1, 900, 300). I will get a Segmentation fault.
The text was updated successfully, but these errors were encountered:
close. #12200 to follow this issue.
Sorry, something went wrong.
No branches or pull requests
I do a test with scatterND:
I found if the shape is (1, 900, 200), the after_mod will success execute. and if I use the shape of (1, 900, 300). I will get a Segmentation fault.
The text was updated successfully, but these errors were encountered: