Add preliminary implementation of WhereAmI()

martin-frbg · Apr 15, 2024 · 34fd140 · 34fd140
1 parent 91a24b8
commit 34fd140
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/common_arm64.h b/common_arm64.h
@@ -55,6 +55,18 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #ifndef ASSEMBLER
 
 
+static __inline int WhereAmI(void){
+  uint64_t ret;
+  __asm__ volatile (
+       "         mrs x0, mpidr_el1 \n"
+       "         and x0, x0, 0xff  \n"
+                 :"=r" (ret)
+                 :: "memory"
+               );
+  if (ret > MAX_CPU_NUMBER) ret = MAX_CPU_NUMBER;
+  return (int)ret;
+}
+
 static __inline void blas_lock(volatile BLASULONG *address){
 
   BLASULONG ret;