When I pass a negative value to the flow.QatConv1d 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.QatConv1d(16, 33, -3, stride=2, quantization_formula="google", quantization_bit=8, quantization_scheme="symmetric")
output = m(input)
System Information
What is your OneFlow installation (pip, source, dockerhub): source
OS: Ubuntu 22.04.3 LTS
OneFlow version (run python3 -m oneflow --doctor):
Summary
When I pass a negative value to the flow.QatConv1d 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
System Information
python3 -m oneflow --doctor
):