SFPI test infra

tenstorrent · Jan 27, 2025 · 33c614a · 33c614a
1 parent 875ce89
commit 33c614a
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 4 deletions.
diff --git a/gcc b/gcc
diff --git a/include/blackhole/sfpi_hw.h b/include/blackhole/sfpi_hw.h
@@ -109,7 +109,7 @@ namespace sfpi {
 
 #elif __GNUC__
 #if !__has_builtin(__builtin_rvtt_bh_sfpload)
-#error TT builtins not found: compile with -mwormhole -march=rv32iw -mabi=ilp32
+#error TT builtins not found: compile with -march=rv32im_xttbh -mabi=ilp32
 #include <fails to compile without extensions>
 #endif // !__has_builtin(__builtin_rvtt_bh_sfpload)
 

diff --git a/include/grayskull/sfpi_hw.h b/include/grayskull/sfpi_hw.h
@@ -91,7 +91,7 @@ namespace sfpi {
 
 #elif __GNUC__
 #if !__has_builtin(__builtin_rvtt_gs_sfpload)
-#error TT builtins not found: compile with -mgrayskull -march=rv32iy -mabi=ilp32
+#error TT builtins not found: compile with -march=rv32i_xttgs -mabi=ilp32
 #include <fails to compile without extensions>
 #endif // !__has_builtin(__builtin_rvtt_gs_sfpload)
 

diff --git a/include/sfpi.h b/include/sfpi.h
@@ -119,10 +119,24 @@
 
 #include <type_traits>
 
+#if __riscv_tt_grayskull
+#define ARCH_GRAYSKULL 1
+#endif
+#if __riscv_tt_wormhole
+#define ARCH_WORMHOLE 1
+#endif
+#if __riscv_tt_blackhole
+#define ARCH_BLACKHOLE 1
+#endif
 #if defined(ARCH_WORMHOLE_B0)
 #define ARCH_WORMHOLE 1
 #endif
 
+#if defined(ARCH_GRAYSKULL) + defined(ARCH_WORMHOLE) + defined(ARCH_BLACKHOLE) != 1
+#error "Exactly one SFPI architecture must be selected"
+#include "stop now, no good will come"
+#endif
+
 #if defined(ARCH_GRAYSKULL)
 #include <grayskull/sfpi_hw.h>
 #elif defined(ARCH_WORMHOLE)

diff --git a/include/wormhole/sfpi_hw.h b/include/wormhole/sfpi_hw.h
@@ -109,7 +109,7 @@ namespace sfpi {
 
 #elif __GNUC__
 #if !__has_builtin(__builtin_rvtt_wh_sfpload)
-#error TT builtins not found: compile with -mwormhole -march=rv32iw -mabi=ilp32
+#error TT builtins not found: compile with -march=rv32im_xttwh -mabi=ilp32
 #include <fails to compile without extensions>
 #endif // !__has_builtin(__builtin_rvtt_wh_sfpload)