Segmentation fault (core dumped) in `flow.nn.MinMaxObserver`

[x0w3n]

Summary

A crash is triggered when the input is a empty tensor.

Code to reproduce bug

import oneflow as flow

invalid_data = flow.tensor([], dtype=flow.float32)
invalid_min_max = flow.nn.MinMaxObserver()
invalid_min_max(invalid_data)

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