Aborted (core dumped) in flow.nn.ReplicationPad2d/ReplicationPad1d

[x0w3n]

Summary

When input some boundary values to oneflow.nn.ReplicationPad2d/ReplicationPad1d's padding parameter, it triggers a crash

Code to reproduce bug

ReplicationPad2d:

import oneflow as flow
flow.nn.ReplicationPad2d(padding=-8353862602220610428)(oneflow.ones((15,5,1,1)))

output:

terminate called after throwing an instance of 'oneflow::RuntimeException'
  what():  Error: CPU can't allocate memory. Tried to allocate 15601624064.0 GB
You can set ONEFLOW_DEBUG or ONEFLOW_PYTHON_STACK_GETTER to 1 to get the Python stack of the error.
Stack trace (most recent call last) in thread 219304:
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1574249, in 
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1573847, in 
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff156f368, in vm::ThreadCtx::TryReceiveAndRun()
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1506818, in vm::EpStreamPolicyBase::Run(vm::Instruction*) const
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff150abe6, in vm::Instruction::Compute()
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1510d2a, in vm::OpCallInstructionPolicy::Compute(vm::Instruction*)
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1510a1f, in 
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff150d182, in 
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7fffed910082, in 

Aborted (Signal sent by tkill() 219060 0)
Aborted (core dumped)

ReplicationPad1d:

import oneflow as flow
oneflow.nn.ReplicationPad1d(padding=-8353862602220610428)(oneflow.ones((15,5)))

output:

terminate called after throwing an instance of 'oneflow::RuntimeException'
  what():  Error: CPU can't allocate memory. Tried to allocate 11275906048.0 GB
You can set ONEFLOW_DEBUG or ONEFLOW_PYTHON_STACK_GETTER to 1 to get the Python stack of the error.
Stack trace (most recent call last) in thread 219952:
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1574249, in 
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1573847, in 
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff156f368, in vm::ThreadCtx::TryReceiveAndRun()
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1506818, in vm::EpStreamPolicyBase::Run(vm::Instruction*) const
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff150abe6, in vm::Instruction::Compute()
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1510d2a, in vm::OpCallInstructionPolicy::Compute(vm::Instruction*)
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff1510a1f, in 
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7ffff150d182, in 
   Object "/home/temp/oneflow-1.0.0/build/liboneflow.so", at 0x7fffed910082, in 

Aborted (Signal sent by tkill() 219711 0)
Aborted (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: ['*/temp/oneflow-1.0.0/python/oneflow']
version: 0.9.1+cpu.git.ec05076
git_commit: ec05076
cmake_build_type: Release
rdma: False
mlir: False

• Python version: 3.10.0
• CUDA driver version: 12.2
• GPU models: NVIDIA GeForce RTX 3090
• Other info: None