oneflow.nn.conv2D causes "core dumped"

[xxxyyyzzz12345]

Summary

The following code snippet produces "Aborted (core dumped)" bug with "dtype=oneflow.float64". However, "dtype=oneflow.float32" won't have this problem.

Code to reproduce bug

import oneflow input = oneflow.rand(4,10,32,32,dtype=oneflow.float64) m = oneflow.nn.Conv2d(10,10,3) output = m(input)

System Information

• What is your OneFlow installation (pip, source, dockerhub): pip
• OS: Ubuntu 20.04.2 LTS
• OneFlow version (run python3 -m oneflow --doctor): 0.7.0+cu112
• Python version: 3.8.8
• CUDA driver version: 11.4

[MARD1NO]

You can change to use this:

import oneflow
input = oneflow.rand(4,10,32,32,dtype=oneflow.float64)
m = oneflow.nn.Conv2d(10,10,3).double()
output = m(input)

.double() will change the parameter to float64 dtype(Default is float32 dtype. ). In most case, we use float32 or float16 in Conv2D