Aborted (core dumped) in flow.nn.AdaptiveMaxPool1d

[x0w3n]

Summary

When I pass a very large value to flow.nn.AdaptiveMaxPool1d , it crashes. It says that the CPU can't allocate memory and tries to allocate a huge memory size. We believe this is a very serious bug that tries to allocate a large amount of memory and could cause a memory overflow vulnerability.

Code to reproduce bug

import oneflow as flow
flow.nn.AdaptiveMaxPool1d(9132760301568586890)(oneflow.randn([2,2,2]))

System Information

• What is your OneFlow installation (pip, source, dockerhub): source
• 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