linalg.generic' op write affecting operations on global resources are restricted to workgroup distributed contexts

[pdhirajkumarprasad]

For the given IR

module {
  func.func @main_graph(%arg0: !torch.vtensor<[1,3,224,224],f32>, %arg1: !torch.vtensor<[1,96,1,1],f32>) ->  !torch.vtensor<[1,32,1,1],f32>  attributes {torch.onnx_meta.ir_version = 8 : si64, torch.onnx_meta.opset_version = 21 : si64, torch.onnx_meta.opset_versions = {ai.onnx.contrib = 1 : si64}, torch.onnx_meta.producer_name = "vai_q_onnx", torch.onnx_meta.producer_version = "1.17.0+5e1d84e"} {
    %52 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<3.125000e-02> : tensor<f32>} : () -> !torch.vtensor<[],f32> 
    %53 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<si8>} : () -> !torch.vtensor<[],si8> 
    %54 = torch.operator "onnx.Constant"() {torch.onnx.value = dense_resource<_onnx__Conv_970_quantized> : tensor<32x24x1x1xsi8>} : () -> !torch.vtensor<[32,24,1,1],si8> 
    %55 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<1.250000e-01> : tensor<f32>} : () -> !torch.vtensor<[],f32> 
    %56 = torch.operator "onnx.Constant"() {torch.onnx.value = dense<0> : tensor<si8>} : () -> !torch.vtensor<[],si8> 
    %57 = torch.operator "onnx.Constant"() {torch.onnx.value = dense_resource<_onnx__Conv_971_quantized> : tensor<32xsi8>} : () -> !torch.vtensor<[32],si8> 
    %1018 = torch.operator "onnx.DequantizeLinear"(%54, %52, %53) : (!torch.vtensor<[32,24,1,1],si8>, !torch.vtensor<[],f32>, !torch.vtensor<[],si8>) -> !torch.vtensor<[32,24,1,1],f32> 
    %1019 = torch.operator "onnx.DequantizeLinear"(%57, %55, %56) : (!torch.vtensor<[32],si8>, !torch.vtensor<[],f32>, !torch.vtensor<[],si8>) -> !torch.vtensor<[32],f32> 
    %1070 = torch.operator "onnx.Conv"(%arg1, %1018, %1019) {torch.onnx.dilations = [1 : si64, 1 : si64], torch.onnx.group = 4 : si64, torch.onnx.kernel_shape = [1 : si64, 1 : si64], torch.onnx.pads = [0 : si64, 0 : si64, 0 : si64, 0 : si64], torch.onnx.strides = [1 : si64, 1 : si64]} : (!torch.vtensor<[1,96,1,1],f32>, !torch.vtensor<[32,24,1,1],f32>, !torch.vtensor<[32],f32>) -> !torch.vtensor<[1,32,1,1],f32> 

    return %1070 : !torch.vtensor<[1,32,1,1],f32>
  }
}

{-#
  dialect_resources: {
    builtin: {
      _onnx__Conv_971_quantized: "0x080000001FE813FD571AFE1C0D1303180E173F05002909E9323E231C1215F610101331F4"
    }
  }
#-}

getting error as

../model.torch_onnx.mlir:11:13: error: 'linalg.generic' op write affecting operations on global resources are restricted to workgroup distributed contexts.
%1070 = torch.operator "onnx.Conv"(%arg1, %1018, %1019) {torch.onnx.dilations = [1 : si64, 1 : si64], torch.onnx.group = 4 : si64, torch.onnx.kernel_shape = [1 : si64, 1 : si64], torch.onnx.pads = [0 : si64, 0 : si64, 0 : si64, 0 : si64], torch.onnx.strides = [1 : si64, 1 : si64]} : (!torch.vtensor<[1,96,1,1],f32>, !torch.vtensor<[32,24,1,1],f32>, !torch.vtensor<[32],f32>) -> !torch.vtensor<[1,32,1,1],f32>
^