[ERROR][/pytorch/catch/torch_mlu/csrc/aten/operators/cnml/internal/maximum_internal.cpp][line:11][cnml_maximum_internal][thread:139997285717824][process:16972]:

Shape of input should match shape of other

[WARNING][/pytorch/catch/torch_mlu/csrc/aten/operators/op_methods.cpp][line:1386][max][thread:139997285717824][process:16972]:

max Op cannot run on MLU device, start running on CPU!

[ERROR][/pytorch/catch/torch_mlu/csrc/aten/operators/cnml/internal/maximum_internal.cpp][line:11][cnml_maximum_internal][thread:139997285717824][process:16972]:

Shape of input should match shape of other

[WARNING][/pytorch/catch/torch_mlu/csrc/aten/operators/op_methods.cpp][line:1386][max][thread:139997285717824][process:16972]:

max Op cannot run on MLU device, start running on CPU!

[WARNING][/pytorch/catch/torch_mlu/csrc/aten/operators/op_methods.cpp][line:3256][convolution_overrideable][thread:139997285717824][process:16972]:

convolution_overrideable Op cannot run on MLU device, start running on CPU!

Traceback (most recent call last):

  File "test_online.py", line 319, in <module>

    test_mlu_mode(args)

  File "test_online.py", line 158, in test_mlu_mode

    out_net = net.forward(img.to(ct.mlu_device())) if (args.mmode != "CPU") else net.forward(img)

  File "/torch/compressai/compressai/models/google.py", line 356, in forward

    y = self.g_a(x)

  File "/torch/venv3/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in __call__

    result = self.forward(*input, **kwargs)

  File "/torch/venv3/pytorch/lib/python3.6/site-packages/torch/nn/modules/container.py", line 92, in forward

    input = module(input)

  File "/torch/venv3/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in __call__

    result = self.forward(*input, **kwargs)

  File "/torch/compressai/compressai/ s/gdn.py", line 83, in forward

    norm = F.conv2d(x**2, gamma, beta)

RuntimeError: To do for CPU