aboutsummaryrefslogtreecommitdiffstats
path: root/classes/unsupportarch.bbclass
diff options
context:
space:
mode:
Diffstat (limited to 'classes/unsupportarch.bbclass')
-rw-r--r--classes/unsupportarch.bbclass11
1 files changed, 11 insertions, 0 deletions
diff --git a/classes/unsupportarch.bbclass b/classes/unsupportarch.bbclass
new file mode 100644
index 0000000..3b8a6b1
--- /dev/null
+++ b/classes/unsupportarch.bbclass
@@ -0,0 +1,11 @@
+BAZEL_TARGET_CPU ??= ""
+BAZEL_TARGET_CPU:x86 = "x86"
+BAZEL_TARGET_CPU:x86-64 = "k8"
+BAZEL_TARGET_CPU:arm = "arm"
+BAZEL_TARGET_CPU:aarch64 = "aarch64"
+
+python __anonymous() {
+ if not d.getVar("BAZEL_TARGET_CPU"):
+ target_arch = d.getVar("TARGET_ARCH")
+ raise bb.parse.SkipPackage("BAZEL_TARGET_CPU is not set\nTensorFlow does not support Target Arch '%s'" % target_arch)
+}