Segmentation fault (core dumped) in `flow.hstack/flow.vstack`

[x0w3n]

Summary

Trying to stack InvalidTensorType tensor triggers a crash.

Code to reproduce bug

import oneflow as flow
x = flow.randn(5, 2)
result = flow.hstack([x, "InvalidTensorType"])

import oneflow as flow
x = flow.randn(5, 2)
result = flow.vstack([x, "InvalidTensorType"])

output:

Segmentation fault (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