Aborted (core dumped)/Segmentation fault (core dumped) in flow.nn.ConstantPad1d/flow.nn.ConstantPad2d

[x0w3n]

Summary

A crash is triggered when the padding parameter is a specific negative number, where flow.nn.ConstantPad1d triggers a Aborted (core dumped), flow.nn.ConstantPad1d triggers Segmentation fault (core dumped).

Code to reproduce bug

Aborted (core dumped)

input_tensor = flow.tensor([[1, 2, 3, 4]], dtype=flow.float32)
padding = -100  # This is invalid padding size
m = flow.nn.ConstantPad1d(padding=padding)
output = m(input_tensor)

Segmentation fault (core dumped)

input_tensor = flow.tensor([[1, 2, 3, 4]], dtype=flow.float32)
padding = -100  # This is invalid padding size
m = flow.nn.ConstantPad2d(padding=padding)
output = m(input_tensor)

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