oneflow
oneflow copied to clipboard
Aborted (core dumped) in `flow.slice_update`
Summary
A crash is triggered when slice_tup_list 's dimension exceeds the boundary.
Code to reproduce bug
import oneflow as flow
import numpy as np
input = flow.tensor(np.array([1, 2, 3]).astype(np.float32))
slice_tup_list = [[2, 2, 1]]
flow.slice_update(input, input, slice_tup_list)
output:
F20241205 09:53:52.836825 2469539 slice_kernel.cpp:260] Check failed: large_slice_param.elem_cnt() == small_slice_param.elem_cnt() (0 vs. 3)
*** Check failure stack trace: ***
@ 0x7fc8fe5d09ca google::LogMessage::Fail()
@ 0x7fc8fe5d0cb2 google::LogMessage::SendToLog()
@ 0x7fc8fe5d0537 google::LogMessage::Flush()
@ 0x7fc8fe5d30a9 google::LogMessageFatal::~LogMessageFatal()
@ 0x7fc8fa1e61f1 oneflow::WriteSlice<>()
@ 0x7fc8fa1f1396 oneflow::SliceUpdateKernel<>::Compute()
@ 0x7fc8fa34e536 oneflow::one::StatefulOpKernel::Compute()
@ 0x7fc8f85e8cab oneflow::vm::OpCallInstructionUtil::Compute()
@ 0x7fc8f85e6787 oneflow::vm::OpCallInstructionPolicy::Compute()
@ 0x7fc8f85e25bc oneflow::vm::Instruction::Compute()
@ 0x7fc8f85e0a6f oneflow::vm::EpStreamPolicyBase::Run()
@ 0x7fc8f85ec086 oneflow::vm::StreamPolicy::RunIf()
@ 0x7fc8f85f36de oneflow::vm::ThreadCtx::TryReceiveAndRun()
@ 0x7fc8f85f5d2d oneflow::(anonymous namespace)::WorkerLoop()
@ 0x7fc8f85f611f _ZNSt6thread11_State_implINS_8_InvokerISt5tupleIJPFvPN7oneflow2vm9ThreadCtxERKSt8functionIFvS6_EEES6_ZNS3_14VirtualMachine15CreateThreadCtxENS3_6SymbolINS3_6DeviceEEENS3_10StreamTypeEmEUlS6_E3_EEEEE6_M_runEv
@ 0x7fc8fe5e540f execute_native_thread_routine
@ 0x7fc9e5f3db43 (unknown)
@ 0x7fc9e5fcfa00 (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