From 4d6746a0ec0efbb473fdc7d682c5e26bfce2021f Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Tue, 28 Nov 2023 00:26:36 +0000 Subject: [PATCH] refactor gelu workload --- .../librispeech_conformer/librispeech_jax/workload.py | 2 +- .../librispeech_conformer/librispeech_pytorch/workload.py | 2 +- algorithmic_efficiency/workloads/workloads.py | 2 +- tests/modeldiffs/librispeech_conformer_gelu/compare.py | 4 ++-- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py index f1267f15c..8745022a8 100644 --- 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 index 18853d19c..f33d75cd4 100644 --- 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 index a972700d3..4231ea9f5 100644 --- 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 index 892040b57..ad695851b 100644 --- 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