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

[Bug] [scatterND] large shape will get a Segmentation fault in x86 #12198

Closed
chengven027 opened this issue Jul 27, 2022 · 1 comment
Closed

Comments

@chengven027
Copy link
Contributor

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.

@chengven027
Copy link
Contributor Author

chengven027 commented Jul 28, 2022

close. #12200 to follow this issue.

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

No branches or pull requests

1 participant