Skip to content

GitLab

Commit dd944f64 authored by Christina Floristean's avatar Christina Floristean
Browse files

Remove AF2 requirement from test

parent f0a320e0
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
tests/test_deepspeed_evo_attention.py
View file @ dd944f64
...@@ -96,15 +96,12 @@ class TestDeepSpeedKernel(unittest.TestCase): ...@@ -96,15 +96,12 @@ class TestDeepSpeedKernel(unittest.TestCase):
self.assertTrue(torch.allclose(torch.abs(out_repro_msa), torch.abs(out_repro_msa_ds), atol=consts.eps)) self.assertTrue(torch.allclose(torch.abs(out_repro_msa), torch.abs(out_repro_msa_ds), atol=consts.eps))
self.assertTrue(torch.allclose(torch.abs(out_repro_pair), torch.abs(out_repro_pair_ds), atol=consts.eps)) self.assertTrue(torch.allclose(torch.abs(out_repro_pair), torch.abs(out_repro_pair_ds), atol=consts.eps))
@compare_utils.skip_unless_alphafold_installed()
def test_compare_evoformer_bf16(self): def test_compare_evoformer_bf16(self):
self.compare_evoformer(torch.bfloat16) self.compare_evoformer(torch.bfloat16)
@compare_utils.skip_unless_alphafold_installed()
def test_compare_evoformer_fp32(self): def test_compare_evoformer_fp32(self):
self.compare_evoformer(torch.float32) self.compare_evoformer(torch.float32)
@compare_utils.skip_unless_alphafold_installed()
def test_dry_run(self): def test_dry_run(self):
with open("tests/test_data/sample_feats.pickle", "rb") as fp: with open("tests/test_data/sample_feats.pickle", "rb") as fp:
batch = pickle.load(fp) batch = pickle.load(fp)
......
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