OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[xla:cpu:onednn] Skip failing tests on Aarch64 CPUs.

Browse Source 
PiperOrigin-RevId: 826675056
This commit is contained in:
Penporn Koanantakool 2025-10-31 16:50:54 -07:00 committed by TensorFlower Gardener
parent 0b5bc94a83
commit 00be2bc09e
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
third_party/xla/xla/service/cpu/tests/BUILD vendored
View File

@ -422,6 +422,7 @@ xla_cc_test(
"//xla/tsl/platform:test", "//xla/tsl/platform:test",
"@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/strings:string_view",
"@com_google_googletest//:gtest_main", "@com_google_googletest//:gtest_main",
"@local_tsl//tsl/platform:platform_port",
], ],
) )

4
third_party/xla/xla/service/cpu/tests/onednn_fusion_test.cc vendored
View File

@ -18,6 +18,7 @@ limitations under the License.
#include "xla/error_spec.h" #include "xla/error_spec.h"
#include "xla/tests/hlo_test_base.h" #include "xla/tests/hlo_test_base.h"
#include "xla/tsl/platform/test.h" #include "xla/tsl/platform/test.h"
#include "tsl/platform/cpu_info.h"
namespace xla::cpu { namespace xla::cpu {
namespace { namespace {
@ -26,7 +27,8 @@ using OneDnnFusionTest = HloTestBase;
inline constexpr bool IsOneDnnGraphEnabled() { inline constexpr bool IsOneDnnGraphEnabled() {
#if defined(XLA_ONEDNN_USE_GRAPH_API) #if defined(XLA_ONEDNN_USE_GRAPH_API)
return true; // Some Aarch64 CPUs have failures. Only test on x86 for now.
return tsl::port::IsX86CPU();
#endif // XLA_ONEDNN_USE_GRAPH_API #endif // XLA_ONEDNN_USE_GRAPH_API
return false; return false;
} }