Summary

When calling flow.dstack(), a type mismatch occurs, resulting in a segmentation fault and a core dump.

Code to reproduce bug

import oneflow as flow
import numpy as np

tensors = [flow.randn(2, 3, 4), np.array([[[1.0, 2.0, 3.0]]])]
tensor_result = flow.dstack(tensors)

output:

Segmentation fault (core dumped)

System Information

• What is your OneFlow installation (pip, source, dockerhub): pip
• OS: Ubuntu 22.04.5 LTS
• OneFlow version (run python3 -m oneflow --doctor):
  path: ['/root/anaconda3/envs/oneflow1.0.0/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 4090
• Other info: None