style change for gelu workload naming

mlcommons · Nov 28, 2023 · 722f1c0 · 722f1c0
1 parent 4d6746a
commit 722f1c0
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py
@@ -380,7 +380,7 @@ def use_post_layer_norm(self) -> bool:
     return False
 
 
-class LibriSpeechConformerGELUWorkload(LibriSpeechConformerWorkload):
+class LibriSpeechConformerGeluWorkload(LibriSpeechConformerWorkload):
 
   @property
   def use_gelu(self) -> bool:

diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py
@@ -346,7 +346,7 @@ def use_post_layer_norm(self) -> bool:
     return False
 
 
-class LibriSpeechConformerGELUWorkload(LibriSpeechConformerWorkload):
+class LibriSpeechConformerGeluWorkload(LibriSpeechConformerWorkload):
 
   @property
   def use_gelu(self) -> bool:

diff --git a/algorithmic_efficiency/workloads/workloads.py b/algorithmic_efficiency/workloads/workloads.py
@@ -58,7 +58,7 @@
     },
     'librispeech_conformer_gelu': {
         'workload_path': 'librispeech_conformer/librispeech',
-        'workload_class_name': 'LibriSpeechConformerGELUWorkload',
+        'workload_class_name': 'LibriSpeechConformerGeluWorkload',
     },
     'librispeech_deepspeech': {
         'workload_path': 'librispeech_deepspeech/librispeech',

diff --git a/tests/modeldiffs/librispeech_conformer_gelu/compare.py b/tests/modeldiffs/librispeech_conformer_gelu/compare.py
@@ -8,9 +8,9 @@
 
 from algorithmic_efficiency import spec
 from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_jax.workload import \
-    LibriSpeechConformerGELUWorkload as JaxWorkload
+    LibriSpeechConformerGeluWorkload as JaxWorkload
 from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.workload import \
-    LibriSpeechConformerGELUWorkload as PytWorkload
+    LibriSpeechConformerGeluWorkload as PytWorkload
 from tests.modeldiffs.diff import out_diff