From 722f1c03020b5ce7e61653aaa1ac2ee312678059 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Tue, 28 Nov 2023 00:32:18 +0000 Subject: [PATCH] style change for gelu workload naming --- .../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 8745022a8..f1267f15c 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 f33d75cd4..18853d19c 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 4231ea9f5..a972700d3 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 ad695851b..892040b57 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