Skip to content

Commit

Permalink
[CPU] fix an assert check for fc tensor parallel (#27922)
Browse files Browse the repository at this point in the history
### Details:
 - *item1*
 - *...*

### Tickets:
 - *ticket-id*
  • Loading branch information
xczhai authored Dec 5, 2024
1 parent 9559b42 commit 29d08ac
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/plugins/intel_cpu/src/nodes/fullyconnected.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ void FullyConnected::prepareParams() {
void FullyConnected::initTensorParallelSync() {
if (tp_cfg.enable_tensor_parallel) {
tp_cfg.id = tp_cfg.sub_memory->get_memory_id(tp_cfg.w_rank);
OPENVINO_ASSERT(tp_cfg.id > 0, "Tensor Parallel Config ID cannot be negative.");
OPENVINO_ASSERT(tp_cfg.id >= 0, "Tensor Parallel Config ID cannot be negative.");
tp_cfg.sub_memory->set_memory_used(tp_cfg.id, tp_cfg.w_rank);
while (true) {
std::lock_guard<std::mutex> lock(tp_cfg.sub_memory->_flagMutex);
Expand Down

0 comments on commit 29d08ac

Please sign in to comment.