From ca54fda3909a0c0c2f5b7f3c510338b0f7894d8b Mon Sep 17 00:00:00 2001
From: Daniel Paoliello <danpao@microsoft.com>
Date: Fri, 23 Feb 2024 10:51:15 -0800
Subject: [PATCH] Support arm64ec

---
 build.rs               | 2 +-
 src/lib.rs             | 2 +-
 testcrate/tests/lse.rs | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/build.rs b/build.rs
index 6268fd2f..b98e4923 100644
--- a/build.rs
+++ b/build.rs
@@ -461,7 +461,7 @@ mod c {
             ]);
         }
 
-        if target_arch == "aarch64" && consider_float_intrinsics {
+        if (target_arch == "aarch64" || target_arch == "arm64ec") && consider_float_intrinsics {
             sources.extend(&[
                 ("__comparetf2", "comparetf2.c"),
                 ("__extenddftf2", "extenddftf2.c"),
diff --git a/src/lib.rs b/src/lib.rs
index 47aef540..191905dd 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -58,7 +58,7 @@ pub mod mem;
 #[cfg(target_arch = "arm")]
 pub mod arm;
 
-#[cfg(target_arch = "aarch64")]
+#[cfg(any(target_arch = "aarch64", target_arch = "arm64ec"))]
 pub mod aarch64;
 
 #[cfg(all(target_arch = "aarch64", target_os = "linux", not(feature = "no-asm"),))]
diff --git a/testcrate/tests/lse.rs b/testcrate/tests/lse.rs
index 5589f22f..e958ef40 100644
--- a/testcrate/tests/lse.rs
+++ b/testcrate/tests/lse.rs
@@ -1,5 +1,5 @@
 #![feature(decl_macro)] // so we can use pub(super)
-#![cfg(all(target_arch = "aarch64", not(feature = "no-asm")))]
+#![cfg(all(any(target_arch = "aarch64", target_arch = "arm64ec"), not(feature = "no-asm")))]
 
 /// Translate a byte size to a Rust type.
 macro int_ty {