add env to support different mm layout on maca.

lmdeploy/pytorch/backends/dlinfer/linear.py

@@ -1,4 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import os
 from typing import Optional
 
 import torch
@@ -11,6 +12,14 @@
 class DlinferLinearImpl(LinearImpl):
     """Dlinfer linear implementation api."""
 
+    def update_weights(self,
+                       weight: torch.Tensor,
+                       bias: Optional[torch.Tensor] = None):
+        """update weights."""
+        if os.getenv('TORCH_MACA_NN_LAYOUT', 'False').lower() == 'false':
+            weight = weight.data.t().contiguous()
+        return weight, bias
+
     def forward(self,
                 x,
                 weight: torch.Tensor,