Skip to content

GitLab

"lib/bindings/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "a745a980dd7cc6361cc96ab2622842410dd80931"
Commit e9898a60 authored by Christina Floristean's avatar Christina Floristean
Browse files

Fix bug in test to enable kernel

parent a4fbd978
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_deepspeed_evo_attention.py
View file @ e9898a60
...@@ -160,7 +160,7 @@ class TestDeepSpeedKernel(unittest.TestCase): ...@@ -160,7 +160,7 @@ class TestDeepSpeedKernel(unittest.TestCase):
out_repro = model(batch) out_repro = model(batch)
# Enable kernel # Enable kernel
model.globals.use_deepspeed_evo_attention = False model.globals.use_deepspeed_evo_attention = True
out_repro_ds = model(batch) out_repro_ds = model(batch)
out_repro = tensor_tree_map(lambda t: t.cpu(), out_repro) out_repro = tensor_tree_map(lambda t: t.cpu(), out_repro)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment