This repository was archived by the owner on Nov 17, 2023. It is now read-only.
This repository was archived by the owner on Nov 17, 2023. It is now read-only.
Ordering operators do not support kNullOp #10085
Description
When testing
a = mx.sym.var("data", shape=(6,7,8))
adata= nd.random.uniform(shape=(6,7,8), ctx=mx.gpu(0))
_, inds = mx.sym.topk(a, axis=-1, k=5, ret_typ='both', is_ascend=False)
exe=inds.bind(mx.gpu(0), {'data':adata})
exe.forward()It saids src/operator/tensor/./ordering_op-inl.h:320: Check failed: req[0] == kWriteTo (0 vs. 1) TopK does not support inplace @sxjscience