Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
@amarin16
amarin16 committed Nov 8, 2024
1 parent 047b98a commit ed6c1e1
Showing 1 changed file with 12 additions and 2 deletions.
14 changes: 12 additions & 2 deletions onnxruntime/test/contrib_ops/skiplayernorm_op_test.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -207,7 +207,12 @@ TEST(SkipLayerNormTest, SkipLayerNormPrePack) {
test.AddInput<MLFloat16>("skip", input_skip_output_dims, ToFloat16({1.f, 1.f, 1.f, 1.f}));
test.AddInput<MLFloat16>("gamma", gamma_beta_bias_dims, ToFloat16({1.f, 1.f}), true);
test.AddInput<MLFloat16>("beta", gamma_beta_bias_dims, ToFloat16({1.f, 1.f}), true);
test.AddOutput<MLFloat16>("output", input_skip_output_dims, ToFloat16({1.f, 1.f, 1.f, 1.f,}));
test.AddOutput<MLFloat16>("output", input_skip_output_dims, ToFloat16({
1.f,
1.f,
1.f,
1.f,
}));

// TRT, DNNL, OpenVINO and NNAPI, CoreML don't support this combination of datatypes
test.Run(OpTester::ExpectResult::kExpectSuccess, "",
Expand All @@ -228,7 +233,12 @@ TEST(SkipLayerNormTest, SkipSimplifiedLayerNormPrePack) {
test.AddInput<MLFloat16>("skip", input_skip_output_dims, ToFloat16({1.f, 1.f, 1.f, 1.f}));
test.AddInput<MLFloat16>("gamma", gamma_beta_bias_dims, ToFloat16({1.f, 1.f}), true);
test.AddInput<MLFloat16>("bias", gamma_beta_bias_dims, ToFloat16({1.f, 1.f}), true);
test.AddOutput<MLFloat16>("output", input_skip_output_dims, ToFloat16({1.f, 1.f, 1.f, 1.f,}));
test.AddOutput<MLFloat16>("output", input_skip_output_dims, ToFloat16({
1.f,
1.f,
1.f,
1.f,
}));

// TRT, DNNL, OpenVINO and NNAPI, CoreML don't support this combination of datatypes
test.Run(OpTester::ExpectResult::kExpectSuccess, "",
Expand Down

0 comments on commit ed6c1e1

Please sign in to comment.