diff --git a/zigzag/classes/stages/SpatialMappingConversionStage.py b/zigzag/classes/stages/SpatialMappingConversionStage.py index 7d08ae4f..617921e7 100644 --- a/zigzag/classes/stages/SpatialMappingConversionStage.py +++ b/zigzag/classes/stages/SpatialMappingConversionStage.py @@ -220,21 +220,12 @@ def convert_user_spatial_mapping(self, user_spatial_mapping): layer=self.layer, accelerator=self.accelerator, ) - try: - SpatialMapping(spatial_mapping_dict=spatial_mapping_dict, layer_node=self.layer), SpatialMapping( - spatial_mapping_dict=spatial_mapping_dict_int, layer_node=self.layer - ) - except: - pass - try: - return SpatialMapping( - spatial_mapping_dict=spatial_mapping_dict, layer_node=self.layer - ), SpatialMapping( - spatial_mapping_dict=spatial_mapping_dict_int, layer_node=self.layer - ) - except: - pass + return SpatialMapping( + spatial_mapping_dict=spatial_mapping_dict, layer_node=self.layer + ), SpatialMapping( + spatial_mapping_dict=spatial_mapping_dict_int, layer_node=self.layer + ) def generate_limited_user_spatial_mapping( self,