Aborted (core dumped) in flow.nn.UpsamplingNearest2d/flow.nn.UpsamplingBilinear2d

[x0w3n]

Summary

When the scale_factor dimension is less than 0, a crash is triggered.

Code to reproduce bug

import oneflow as flow
import numpy as np

input_shape = (1, 1, 2, 2)
input = flow.tensor(np.arange(1, 5).reshape((1, 1, 2, 2)), dtype=flow.float32)
scale_factor = (-1.0, 0.5)

m = flow.nn.UpsamplingBilinear2d(scale_factor=scale_factor)
output = m(input)
output.mean()

import oneflow as flow
import numpy as np

input_shape = (1, 1, 2, 2)
input = flow.tensor(np.arange(1, 5).reshape((1, 1, 2, 2)), dtype=flow.float32)

scale_factor = (-1.0, 0.5)

m = flow.nn.UpsamplingNearest2d(scale_factor=scale_factor)
output = m(input)
output.mean()

output:

F20241205 09:31:51.857672 2449429 bin_allocator.h:219] Check failed: ptr2piece_.emplace(piece->ptr, piece).second 
*** Check failure stack trace: ***
    @     0x7f6aee9d09ca  google::LogMessage::Fail()
    @     0x7f6aee9d0cb2  google::LogMessage::SendToLog()
    @     0x7f6aee9d0537  google::LogMessage::Flush()
    @     0x7f6aee9d30a9  google::LogMessageFatal::~LogMessageFatal()
    @     0x7f6ae89de860  oneflow::vm::BinAllocator<>::MarkPiece()
    @     0x7f6ae89df4ad  oneflow::vm::BinAllocator<>::FindPiece()
    @     0x7f6ae89df670  oneflow::vm::BinAllocator<>::Allocate()
    @     0x7f6ae89e8f61  oneflow::vm::OpCallInstructionUtil::Compute()
    @     0x7f6ae89e6787  oneflow::vm::OpCallInstructionPolicy::Compute()
    @     0x7f6ae89e25bc  oneflow::vm::Instruction::Compute()
    @     0x7f6ae89e0a6f  oneflow::vm::EpStreamPolicyBase::Run()
    @     0x7f6ae89ec086  oneflow::vm::StreamPolicy::RunIf()
    @     0x7f6ae89f36de  oneflow::vm::ThreadCtx::TryReceiveAndRun()
    @     0x7f6ae89f5d18  oneflow::(anonymous namespace)::WorkerLoop()
    @     0x7f6ae89f611f  _ZNSt6thread11_State_implINS_8_InvokerISt5tupleIJPFvPN7oneflow2vm9ThreadCtxERKSt8functionIFvS6_EEES6_ZNS3_14VirtualMachine15CreateThreadCtxENS3_6SymbolINS3_6DeviceEEENS3_10StreamTypeEmEUlS6_E3_EEEEE6_M_runEv
    @     0x7f6aee9e540f  execute_native_thread_routine
    @     0x7f6bd63b0b43  (unknown)
    @     0x7f6bd6442a00  (unknown)
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: ['/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