Aborted (core dumped) in oneflow.nn.functional.avg_pool1d - 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.87k stars 667 forks source link

Aborted (core dumped) in oneflow.nn.functional.avg_pool1d #10506

Open x0w3n opened 5 months ago

x0w3n commented 5 months ago

Summary

flow.nn.avg_pool1d crashes when its input is an empty tensor.

Code to reproduce bug

import oneflow as flow
oneflow.nn.functional.avg_pool1d(input=oneflow.ones([]), kernel_size=1, stride=1)

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: ['*/temp/oneflow-1.0.0/python/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