mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add smoketest for autodidax

Browse Source
This commit is contained in:
Jake VanderPlas 2023-01-27 08:18:01 -08:00
parent 87051ba875
commit c89b537f3a
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/api_test.py
View File

@ -20,7 +20,9 @@ import copy
import enum
from functools import partial
import inspect
import importlib
import operator
import os
import platform
import re
import subprocess
@ -9509,5 +9511,17 @@ class EnvironmentInfoTest(jtu.JaxTestCase):
assert f"jaxlib: {lib.version_str}" in result
assert f"numpy: {np.__version__}" in result
class AutodidaxTest(jtu.JaxTestCase):
def test_autodidax_smoketest(self):
autodidax_file = os.path.join(
os.path.dirname(os.path.dirname(__file__)),
'docs',
'autodidax.py')
if not os.path.exists(autodidax_file):
self.skipTest("Cannot locate autodidax.py")
spec = importlib.util.spec_from_file_location('autodidax', 'docs/autodidax.py')
autodidax_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(autodidax_module)
if __name__ == '__main__':
absltest.main(testLoader=jtu.JaxTestLoader())