Skip to content

Commit 0bbe6ac

Browse files
authored
bugfix: fix build break on mlu device. (#510)
Signed-off-by: pengtao.156 <pengtao.156@jd.com>
1 parent 2dd568e commit 0bbe6ac

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

xllm/core/layers/common/word_embedding_impl.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ limitations under the License.
1515

1616
#pragma once
1717

18-
#include <cuda_runtime.h>
1918
#include <glog/logging.h>
2019
#include <torch/torch.h>
2120

xllm/core/runtime/acl_graph_executor_impl.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@ limitations under the License.
2626
#include "core/framework/kv_cache/kv_cache.h"
2727
#include "core/framework/model/causal_lm.h"
2828
#include "core/framework/model/model_input_params.h"
29-
// #include "core/layers/attention_mask.h"
3029
#include "executor_impl.h"
3130
#include "options.h"
3231
#include "torch_npu/csrc/core/npu/NPUGraph.h"

0 commit comments

Comments
 (0)