Skip to content

Commit 10aed66

Browse files
authored
Merge pull request fastmachinelearning#577 from fastmachinelearning/jmitrevs/layer_name_case
Make layer names case sensitive in config.
2 parents bf33047 + f06f28e commit 10aed66

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hls4ml/model/graph.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def get_output_dir(self):
5858
def get_layer_config_value(self, layer, key, default=None):
5959
hls_config = self.config['HLSConfig']
6060

61-
name_config = hls_config.get('LayerName', {}).get(layer.name.lower(), None)
61+
name_config = hls_config.get('LayerName', {}).get(layer.name, None)
6262
if name_config is not None:
6363
return name_config.get(key, default)
6464

@@ -80,7 +80,7 @@ def get_layer_config(self, layer):
8080
if type_config is not None:
8181
layer_config.update(type_config)
8282

83-
name_config = hls_config.get('LayerName', {}).get(layer.name.lower(), None)
83+
name_config = hls_config.get('LayerName', {}).get(layer.name, None)
8484
if name_config is not None:
8585
layer_config.update(name_config)
8686

0 commit comments

Comments
 (0)