Skip to content

Commit 98a0be1

Browse files
committed
issue/900 - adjust doc text and test script
1 parent bea5068 commit 98a0be1

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

test/infinicore/nn/HOW_TO_USE_GRAPH_RECORDING_TEST.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
### 运行测试
66

77
```bash
8-
cd /home/zhuyue/codes/InfiniCore
8+
cd <path_to>/InfiniCore
99
python test/infinicore/nn/test_embedding_graph_recording.py
1010
```
1111

@@ -259,7 +259,7 @@ python test/infinicore/nn/test_embedding_graph_recording.py
259259
#!/bin/bash
260260
# quick_check.sh
261261

262-
cd /home/zhuyue/codes/InfiniCore
262+
cd <path_to>/InfiniCore
263263

264264
echo "=== 1. 代码检查 ==="
265265
if grep -q "to(cpu_device)" src/infinicore/nn/embedding.cc; then

test/infinicore/nn/test_embedding_graph_recording.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515

1616
import infinicore
1717
import torch
18-
import ctypes
1918

2019

2120
def test_embedding_graph_recording():
@@ -73,7 +72,7 @@ def test_embedding_graph_recording():
7372
warmup_input = input_ids_device
7473

7574
# Warmup(图录制前需要先执行一次,包括内存分配)
76-
warmup_output = embedding.forward(warmup_input)
75+
embedding.forward(warmup_input)
7776
infinicore.sync_stream() # 同步确保 warmup 完成
7877

7978
# 预先分配输出张量(CUDA Graph 不支持动态内存分配)

0 commit comments

Comments
 (0)