Aborted (core dumped) in flow.unfold

[x0w3n]

Summary

A crash is triggered when the processed tensor shape is incompatible with the dimension.

Code to reproduce bug

import oneflow as flow

x = flow.arange(1, 8)
dimension = len(x.shape)
print("dimension:",dimension)
result = x.unfold(dimension, 1, 2)

output:

dimension:1
F20241205 10:44:16.174775 2497410 shape.cpp:30] Check failed: index < tp()->NumAxes() (1 vs. 1)  Shape: (7,) visit index: 1 > num_axes: 1
*** Check failure stack trace: ***
    @     0x7fd20d1d09ca  google::LogMessage::Fail()
    @     0x7fd20d1d0cb2  google::LogMessage::SendToLog()
    @     0x7fd20d1d0537  google::LogMessage::Flush()
    @     0x7fd20d1d30a9  google::LogMessageFatal::~LogMessageFatal()
    @     0x7fd202acba57  oneflow::ConstShapeMixIn<>::At()
    @     0x7fd206178ba4  oneflow::one::view::UnfoldTensor()
    @     0x7fd206288444  _ZNSt17_Function_handlerIFN7oneflow5MaybeINS0_3one6TensorEvEERKSt10shared_ptrIS3_ERKiSA_SA_EZNS2_10functional18PackedFunctorMakerISB_E4makeINSC_4impl19UnfoldTensorFunctorELi0EEENSC_13PackedFunctorISB_EERKSsRKT_EUlS8_SA_SA_SA_E_E9_M_invokeERKSt9_Any_dataS8_SA_SA_SA_
    @     0x7fd209604f6a  oneflow::one::functional::UnfoldTensor()
    @     0x7fd2ec610632  oneflow::one::functional::unfold_tensor()
    @     0x7fd2ec66b4c2  (unknown)
    @           0x4f9b46  method_vectorcall_VARARGS_KEYWORDS
    @           0x4e7f6d  _PyEval_EvalFrameDefault
    @           0x4e69da  _PyEval_EvalCode
    @           0x4e6667  _PyEval_EvalCodeWithName
    @           0x4e6619  PyEval_EvalCodeEx
    @           0x5938eb  PyEval_EvalCode
    @           0x5c1157  run_eval_code_obj
    @           0x5bd170  run_mod
    @           0x456423  pyrun_file.cold
    @           0x5b6e52  PyRun_SimpleFileExFlags
    @           0x5b43ce  Py_RunMain
    @           0x587999  Py_BytesMain
    @     0x7fd2f4c71d90  (unknown)
    @     0x7fd2f4c71e40  __libc_start_main
    @           0x58784e  (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