Try a different formula, see if it makes PLAIDML_CPU happy

768c196d · Adam Procter · d485b09f · 768c196d
Commit 768c196d authored Aug 01, 2019 by Adam Procter
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

plaidml_ops_logical.cpp src/ngraph/runtime/plaidml/plaidml_ops_logical.cpp +1 -1

No files found.
--- a/src/ngraph/runtime/plaidml/plaidml_ops_logical.cpp
+++ b/src/ngraph/runtime/plaidml/plaidml_ops_logical.cpp
@@ -81,6 +81,6 @@ void ngraph::runtime::plaidml::ImplXor::Apply()
                   .add(builder::Input{op_input(0), "A"})
                   .add(builder::Input{op_input(1), "B"})
                   .add(builder::Output{"C"})
-                   .add(builder::Elementwise{"C", "A ? cmp_eq(B, 0) : B"})
+                   .add(builder::Elementwise{"C", "A ? (B ? 0 : A) : B"})
                   .finalize());
 }