Remove redundant code in ColumnParallelLinear (vllm-project#146)

g-i-o-r-g-i-o · Jun 11, 2023 · da5ddcd · da5ddcd
1 parent 5020e1e
commit da5ddcd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cacheflow/model_executor/parallel_utils/tensor_parallel/layers.py b/cacheflow/model_executor/parallel_utils/tensor_parallel/layers.py
@@ -305,7 +305,7 @@ def forward(self, input_):
         """
         bias = self.bias if not self.skip_bias_add else None
 
-        input_parallel = copy_to_tensor_model_parallel_region(input_)
+        input_parallel = input_
         # Matrix multiply.
         output_parallel = F.linear(input_parallel, self.weight, bias)
         if self.gather_output: