Aborted (core dumped) in flow.tensor_scatter_nd_update

[x0w3n]

Summary

A crash is triggered when the indices parameter exceeds the dimensions of the input tensor.

Code to reproduce bug

import oneflow as flow

tensor = flow.tensor([0, 1, 2, 3, 4], dtype=flow.int64)
indices = flow.tensor([[9], [1]], dtype=flow.int64)
updates = flow.tensor([-1,-2], dtype=flow.int64)
result = flow.tensor_scatter_nd_update(tensor, indices, updates)

output:

F20241205 09:48:30.094549 2465046 nd_index_slice_util.h:103] Check failed: nd_index[i] < dense_shape[i] && nd_index[i] >= -dense_shape[i] IndexError: index 9 is out of bounds for dimension 0 with size 5
*** Check failure stack trace: ***
    @     0x7f0e055d09ca  google::LogMessage::Fail()
    @     0x7f0e055d0cb2  google::LogMessage::SendToLog()
    @     0x7f0e055d0537  google::LogMessage::Flush()
    @     0x7f0e055d30a9  google::LogMessageFatal::~LogMessageFatal()
    @     0x7f0e00ba0707  oneflow::OffsetInSliceToOffsetInDense<>()
    @     0x7f0e00ba1c97  oneflow::TensorScatterNdUpdateKernel<>::Compute()
    @     0x7f0e0134e54d  oneflow::one::StatefulOpKernel::Compute()
    @     0x7f0dff5e8cab  oneflow::vm::OpCallInstructionUtil::Compute()
    @     0x7f0dff5e6787  oneflow::vm::OpCallInstructionPolicy::Compute()
    @     0x7f0dff5e25bc  oneflow::vm::Instruction::Compute()
    @     0x7f0dff5e0a6f  oneflow::vm::EpStreamPolicyBase::Run()
    @     0x7f0dff5ec086  oneflow::vm::StreamPolicy::RunIf()
    @     0x7f0dff5f36de  oneflow::vm::ThreadCtx::TryReceiveAndRun()
    @     0x7f0dff5f5d18  oneflow::(anonymous namespace)::WorkerLoop()
    @     0x7f0dff5f611f  _ZNSt6thread11_State_implINS_8_InvokerISt5tupleIJPFvPN7oneflow2vm9ThreadCtxERKSt8functionIFvS6_EEES6_ZNS3_14VirtualMachine15CreateThreadCtxENS3_6SymbolINS3_6DeviceEEENS3_10StreamTypeEmEUlS6_E3_EEEEE6_M_runEv
    @     0x7f0e055e540f  execute_native_thread_routine
    @     0x7f0eecf84b43  (unknown)
    @     0x7f0eed016a00  (unknown)
Aborted (core dumped)

System Information

• What is your OneFlow installation (pip, source, dockerhub): pip
• OS: Ubuntu 22.04.3 LTS
• OneFlow version (run python3 -m oneflow --doctor):

path: ['/home/miniconda3/envs/oneflow/lib/python3.9/site-packages/oneflow']
version: 0.9.0
git_commit: 381b12c
cmake_build_type: Release
rdma: True
mlir: True

• Python version: 3.9.13
• CUDA driver version: 12.2
• GPU models: NVIDIA GeForce RTX 4090
• Other info: None