Aborted (core dumped) in flow.nn.functional.conv3d

[x0w3n]

Summary

it triggers a crash when I pass a multi-dimensional tensor to the input parameter and the weight parameter.

Code to reproduce bug

import oneflow as flow
oneflow.nn.functional.conv3d(input=flow.ones([1,1,1,1]), weight=flow.ones([1,1,1,1]))

Relevant log output

F20240412 05:47:20.321039 4113438 shape.cpp:30] Check failed: index < tp()->NumAxes() (2 vs. 0)  Shape: () visit index: 2 > num_axes: 0
*** Check failure stack trace: ***
    @     0x7fff133d09ca  google::LogMessage::Fail()
    @     0x7fff133d0cb2  google::LogMessage::SendToLog()
    @     0x7fff133d0537  google::LogMessage::Flush()
    @     0x7fff133d30a9  google::LogMessageFatal::~LogMessageFatal()
    @     0x7fff08ccba57  oneflow::ConstShapeMixIn<>::At()
    @     0x7fff0c751336  oneflow::one::functional::impl::ConvBaseFunctor::operator()()
    @     0x7fff0c751a8c  _ZNSt17_Function_handlerIFN7oneflow5MaybeINS0_3one6TensorEvEERKSt10shared_ptrIS3_ES8_RKNS0_8OptionalIS3_EERKSt6vectorIiSaIiEESH_SH_RKiRKSsEZNS2_10functional18PackedFunctorMakerISM_E4makeINSN_4impl13Conv1dFunctorELi0EEENSN_13PackedFunctorISM_EESL_RKT_EUlS8_S8_SC_SH_SH_SH_SJ_SL_E_E9_M_invokeERKSt9_Any_dataS8_S8_SC_SH_SH_SH_SJ_SL_    @     0x7fff0f8aae5c  oneflow::one::functional::Conv1d()
    @     0x7fffeff8d07f  oneflow::one::functional::conv1d()
    @     0x5555556c6df4  cfunction_call
    @     0x55555568dd1e  _PyObject_MakeTpCall.localalias.3
    @     0x555555731715  _PyEval_EvalFrameDefault
    @     0x5555556db1eb  _PyEval_Vector
    @     0x5555556ec0e8  PyEval_EvalCode
    @     0x55555579cf1b  run_eval_code_obj
    @     0x5555557d2805  run_mod
    @     0x5555557d4bb0  pyrun_file
    @     0x5555557d4d98  _PyRun_SimpleFileObject.localalias.11
    @     0x5555557d4ea4  _PyRun_AnyFileObject.localalias.17
    @     0x5555557d605f  Py_RunMain.localalias.0
    @     0x5555557d61e9  Py_BytesMain
    @     0x7ffff7cbcd90  (unknown)
    @     0x7ffff7cbce40  __libc_start_main
    @     0x55555575ae86  (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: ['/root/anaconda3/envs/oneflow/lib/python3.10/site-packages/oneflow']
  version: 0.9.0
  git_commit: 381b12c
  cmake_build_type: Release
  rdma: True
  mlir: True

• Python version: 3.10.0
• CUDA driver version: 12.2
• GPU models: NVIDIA GeForce RTX 3090
• Other info: None