Skip to content

Commit b5f0eaa

Browse files
committed
chore: Cleanup of extra comments
Signed-off-by: Brandon Groth <[email protected]>
1 parent b00287f commit b5f0eaa

File tree

4 files changed

+0
-12
lines changed

4 files changed

+0
-12
lines changed

fms_mo/dq.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,6 @@
4646
from fms_mo.utils.aiu_utils import save_for_aiu
4747
from fms_mo.utils.dq_utils import config_quantize_smooth_layers
4848
from fms_mo.utils.eval_utils import Evaluator, eval_llm_1GPU
49-
50-
# from fms_mo.utils.import_utils import available_packages
5149
from fms_mo.utils.utils import patch_torch_bmm, prepare_input
5250

5351
logger = logging.getLogger(__name__)

fms_mo/fx/utils.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -172,8 +172,6 @@ def lower_qmodel_to_ext_kernels(
172172
QLinearExllamaV2,
173173
)
174174

175-
# qclass_accepted = list( qcfg["mapping"].values() )
176-
177175
mod2swap = {
178176
n: m
179177
for n, m in mod.named_modules()

fms_mo/utils/qconfig_utils.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1285,7 +1285,6 @@ def check_config(config: dict, model_dtype: torch.dtype = None) -> None:
12851285
mx_spec_int_var_str_defaults = [
12861286
("scale_bits", 8),
12871287
("block_size", 32),
1288-
# ("fp", 16), # can only set either fp or bfloat to non-zero
12891288
("bfloat", 16),
12901289
]
12911290
mx_spec_int_var_values = {2, 4, 6, 8, 16, 32}
@@ -1388,4 +1387,3 @@ def check_config(config: dict, model_dtype: torch.dtype = None) -> None:
13881387
else:
13891388
if qbmm_map is not QBmm:
13901389
raise ValueError("Mapping for matmul_or_bmm is not QBmm")
1391-
# End mx_specs checks

tests/models/conftest.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -838,7 +838,6 @@ def config_fp32_mx(request):
838838
qconfig = deepcopy(request.param)
839839
mx_specs = get_mx_specs_defaults()
840840

841-
# print(f"fixture qcfg before {qconfig=}")
842841
# Set config vars prefixed w/ "mx_"
843842
for key, val in mx_specs.items():
844843
qconfig["mx_" + key] = val
@@ -854,11 +853,6 @@ def config_fp32_mx(request):
854853
del qconfig["mx_a_elem_format"]
855854
del qconfig["mx_w_elem_format"]
856855

857-
# Set mx_specs as if we ran qconfig_init
858-
# set_mx_specs(qconfig)
859-
860-
# print(f"fixture qcfg after", qconfig["mx_specs"])
861-
862856
return qconfig
863857

864858

0 commit comments

Comments
 (0)