Skip to content

Commit 4012888

Browse files
committed
Adjust activation support check to follow new Neutron-C flow requirements
1 parent a6ee309 commit 4012888

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

backends/nxp/backend/neutron_operator_support.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,13 +86,17 @@ def transposition_is_supported_on_neutron(
8686

8787

8888
def activation_supported_on_target(
89-
node: Node, neutron_target_spec: NeutronTargetSpec
89+
node: Node, neutron_target_spec: NeutronTargetSpec, use_new_flow_neutron_c: bool = False,
9090
) -> bool:
9191
"""This function determines if the current NeutronSoftware properly supports an activation operator represented by the given node.
9292
9393
:param node: The node representing the activation operator.
9494
:param neutron_target_spec: Object for querying the target platform to retrieve its properties.
9595
"""
96+
97+
if use_new_flow_neutron_c:
98+
return True
99+
96100
input_shape = list(input_tensor(node, 0).shape)
97101
if node.args[0].meta[NXP_NODE_FORMAT].is_channels_first():
98102
input_shape = dims_to_channels_last(input_shape)

0 commit comments

Comments
 (0)