Skip to content

Commit c4f0455

Browse files
author
Yi Zhang
committed
update
1 parent 3d75696 commit c4f0455

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

onnxruntime/test/providers/xnnpack/xnnpack_basic_test.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ TEST(XnnpackEP, TestNhwcConvReluClipFusion) {
9090
RunAndVerifyOutputsWithEP(ort_model_path, "TestNhwcConvReluClipFusion", std::move(ep), feeds, params);
9191
}
9292

93+
#ifdef XNNPACK_FP16_SUPPORTED
9394
TEST(XnnpackEP, TestNhwcConvReluClipFusion_FP16) {
9495
const ORTCHAR_T* ort_model_path = ORT_MODEL_FOLDER "nhwc_conv_clip_relu_fp16.onnx";
9596

@@ -124,16 +125,15 @@ TEST(XnnpackEP, TestNhwcConvReluClipFusion_FP16) {
124125
};
125126

126127
EPVerificationParams params;
127-
// TODO: make it to ExpectedEPNodeAssignment::All if more Fp16 ops are added.
128-
// Now, set it to Some to pass the test in iOS pipeline
129-
params.ep_node_assignment = ExpectedEPNodeAssignment::Some;
128+
params.ep_node_assignment = ExpectedEPNodeAssignment::All;
130129
params.fp32_abs_err = 0.0002f;
131130
params.graph_verifier = &verify;
132131

133132
auto ep = DefaultXnnpackExecutionProvider();
134133
// So far, CPU EP doensn't support Fp16 Conv fusion, so verify_outputs is skipped.
135134
RunAndVerifyOutputsWithEP(ort_model_path, "TestNhwcConvReluClipFusion_FP16", std::move(ep), feeds, params, {}, false);
136135
}
136+
#endif
137137

138138
// test we can share the cpu ep allocator with the xnnpack EP
139139
TEST(XnnpackEP, TestAllocatorSharing) {

0 commit comments

Comments
 (0)