Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Aborted (core dumped) in flow.QatConv2d #10509

Open
x0w3n opened this issue Apr 23, 2024 · 0 comments · May be fixed by #10532
Open

Aborted (core dumped) in flow.QatConv2d #10509

x0w3n opened this issue Apr 23, 2024 · 0 comments · May be fixed by #10532
Labels
bug community events from community

Comments

@x0w3n
Copy link

x0w3n commented Apr 23, 2024

Summary

When I pass a negative value to the flow.QatConv2d parameter, it crashes. 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 lead to a memory overflow vulnerability.

Code to reproduce bug

import oneflow as flow
import numpy as np
arr = np.random.randn(20, 16, 50)
input = flow.Tensor(arr)
m = flow.nn.QatConv2d(-16, 33, 3, stride=2, quantization_formula="google", quantization_bit=8, quantization_scheme="symmetric")
output = m(input)

image

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.f97cd07
git_commit: f97cd07
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
@x0w3n x0w3n added bug community events from community labels Apr 23, 2024
@Dmovic Dmovic linked a pull request May 30, 2024 that will close this issue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug community events from community
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant