Aborted (core dumped) in flow.cumsum/flow.cumprod

[x0w3n]

Summary

crash is triggered when dim=0.

Code to reproduce bug

import oneflow as flow

x = flow.tensor(5)
result = x.cumprod(dim=0)

import oneflow as flow

x = flow.tensor(5)
result = x.cumsum(dim=0)

output:

F20241205 10:06:16.030812 2476803 shape.cpp:30] Check failed: index < tp()->NumAxes() (0 vs. 0)  Shape: () visit index: 0 > num_axes: 0
*** Check failure stack trace: ***
    @     0x7f5412fd09ca  google::LogMessage::Fail()
    @     0x7f5412fd0cb2  google::LogMessage::SendToLog()
    @     0x7f5412fd0537  google::LogMessage::Flush()
    @     0x7f5412fd30a9  google::LogMessageFatal::~LogMessageFatal()
    @     0x7f54088cbddc  oneflow::ConstShapeMixIn<>::At()
    @     0x7f540d410b00  oneflow::CpuCumKernel<>::Compute()
    @     0x7f540ed4e54d  oneflow::one::StatefulOpKernel::Compute()
    @     0x7f540cfe8cab  oneflow::vm::OpCallInstructionUtil::Compute()
    @     0x7f540cfe6787  oneflow::vm::OpCallInstructionPolicy::Compute()
    @     0x7f540cfe25bc  oneflow::vm::Instruction::Compute()
    @     0x7f540cfe0a6f  oneflow::vm::EpStreamPolicyBase::Run()
    @     0x7f540cfec086  oneflow::vm::StreamPolicy::RunIf()
    @     0x7f540cff36de  oneflow::vm::ThreadCtx::TryReceiveAndRun()
    @     0x7f540cff5d18  oneflow::(anonymous namespace)::WorkerLoop()
    @     0x7f540cff611f  _ZNSt6thread11_State_implINS_8_InvokerISt5tupleIJPFvPN7oneflow2vm9ThreadCtxERKSt8functionIFvS6_EEES6_ZNS3_14VirtualMachine15CreateThreadCtxENS3_6SymbolINS3_6DeviceEEENS3_10StreamTypeEmEUlS6_E3_EEEEE6_M_runEv
    @     0x7f5412fe540f  execute_native_thread_routine
    @     0x7f54fa99bb43  (unknown)
    @     0x7f54faa2da00  (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