Aborted (core dumped) in flow.repeat_interleave

[x0w3n]

Summary

This will raise a crash because a 0D tensor cannot be repeated meaningfully.

Code to reproduce bug

import oneflow as flow

zero_dim_tensor = flow.tensor(5)
result = zero_dim_tensor.repeat_interleave(3)

output:

F20241205 10:27:51.987188 2488225 shape.cpp:30] Check failed: index < tp()->NumAxes() (0 vs. 0)  Shape: () visit index: 0 > num_axes: 0
*** Check failure stack trace: ***
    @     0x7f043fbd09ca  google::LogMessage::Fail()
    @     0x7f043fbd0cb2  google::LogMessage::SendToLog()
    @     0x7f043fbd0537  google::LogMessage::Flush()
    @     0x7f043fbd30a9  google::LogMessageFatal::~LogMessageFatal()
    @     0x7f04354cba57  oneflow::ConstShapeMixIn<>::At()
    @     0x7f0438ccb20e  oneflow::one::functional::impl::RepeatInterLeaveIntFunctor::operator()()
    @     0x7f0438ccc809  _ZNSt17_Function_handlerIFN7oneflow5MaybeINS0_3one6TensorEvEERKSt10shared_ptrIS3_ERKiRKNS0_8OptionalIiEEEZNS2_10functional18PackedFunctorMakerISF_E4makeINSG_4impl26RepeatInterLeaveIntFunctorELi0EEENSG_13PackedFunctorISF_EERKSsRKT_EUlS8_SA_SE_E_E9_M_invokeERKSt9_Any_dataS8_SA_SE_
    @     0x7f043c0a284d  oneflow::one::functional::RepeatInterLeaveInt()
    @     0x7f051f185a88  oneflow::one::functional::repeat_interleave()
    @           0x507397  cfunction_call
    @           0x505888  PyObject_Call
    @           0x4ed555  _PyEval_EvalFrameDefault
    @           0x4e69da  _PyEval_EvalCode
    @           0x4f7de4  _PyFunction_Vectorcall
    @           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
    @     0x7f0527604d90  (unknown)
    @     0x7f0527604e40  __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