apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

rename fused_get_rope.cu (#3752)

Browse Source 
* rename fused_get_rope.cu

* fix

* fix typos

* fix

* fix
This commit is contained in:
co63oc
2025-09-03 10:54:34 +08:00
committed by GitHub
parent 2c9b169c0e
commit 5441538173
17 changed files with 31 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/graph_optimization/test_cuda_graph_spec_decode.py
View File

@@ -117,14 +117,14 @@ class TestCUDAGrpahSpecDecode(unittest.TestCase):
input_tensor1 = paddle.ones([1, 32768])
forward_meta1 = ForwardMeta(input_ids=input_tensor1, ids_remove_padding=input_tensor1, step_use_cudagraph=True)
# Triger Capture
# Trigger Capture
_ = test_model1(ids_remove_padding=input_tensor1, forward_meta=forward_meta1)
# Reaplay
# Replay
_ = test_model1(ids_remove_padding=input_tensor1, forward_meta=forward_meta1)
output1 = test_model1(ids_remove_padding=input_tensor1, forward_meta=forward_meta1)
# Corrent output
# Correct output
output1_correct = test_model1.forward_correct(ids_remove_padding=input_tensor1, forward_meta=forward_meta1)
assert (output1 == output1_correct).all()