Skip to content

Commit

Permalink
staticlize check qwen
Browse files Browse the repository at this point in the history
  • Loading branch information
zhen8838 committed Jan 2, 2025
1 parent 9066b08 commit 1f92ef1
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion src/Nncase.Passes/Rules/ShapeBucket/ShapeBucket.cs
Original file line number Diff line number Diff line change
Expand Up @@ -472,7 +472,7 @@ public MatmulToFusion(bool isDynamic = false)
{
}

public override bool Check(Call call)
public static bool CheckQWEN(Call call)
{
if (System.Environment.GetEnvironmentVariable("NNCASE_QWEN_QUANT_LAYERS") is string quant_layers)
{
Expand Down Expand Up @@ -537,6 +537,11 @@ public override bool Check(Call call)

return true;
}

public override bool Check(Call call)
{
return CheckQWEN(call);
}
}

public class ActToFusion : MarkerCallToFusion<ActivationOp>
Expand Down

0 comments on commit 1f92ef1

Please sign in to comment.