full name for neuron

patrick-toulme · Dec 11, 2024 · 52d0991 · 52d0991
1 parent 1bdb44e
commit 52d0991
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/axlearn/common/flash_attention/neuron_attention_test.py b/axlearn/common/flash_attention/neuron_attention_test.py
@@ -12,7 +12,7 @@
 
 
 if jax.default_backend() != "neuron":
-    pytestmark = pytest.mark.skip(reason="Incompatible hardware, Neuron only test.")
+    pytestmark = pytest.mark.skip(reason="Incompatible hardware, AWS Neuron only test.")
 
 
 @pytest.mark.parametrize(