Aborted (core dumped) in flow.nn.functional.conv2d - Githubissues

Oneflow-Inc / oneflow

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.

http://www.oneflow.org

Apache License 2.0

5.79k stars 658 forks source link

Aborted (core dumped) in flow.nn.functional.conv2d #10488

Open x0w3n opened 2 months ago

x0w3n commented 2 months ago

Summary

it triggers a crash when I pass a multi-dimensional tensor to flow.nn.functional.conv2d.

Code to reproduce bug

mport oneflow as flow
a = flow.ones([1,1,1,1])
b = flow.tensor(1)
print(a)
print(b)
flow.nn.functional.conv2d(a,b)

Relevant log output

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