test: remove gpu memory allocation

to run the test by Github actions
tensorops · May 30, 2023 · 4a4ada7 · 4a4ada7
1 parent 504a2fc
commit 4a4ada7
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/tests/layers/test_addnorm.py b/tests/layers/test_addnorm.py
@@ -4,9 +4,6 @@
 
 from transformerx.layers import AddNorm
 
-physical_devices = tf.config.list_physical_devices("GPU")
-tf.config.experimental.set_memory_growth(physical_devices[0], True)
-
 
 class TestAddNorm:
     def test_init(self):