diff --git a/BUILD b/BUILD index 89a33095dd..1107a7b75e 100644 --- a/BUILD +++ b/BUILD @@ -114,6 +114,10 @@ COPTS = select({ "-march=rv64gcv1p0", "-menable-experimental-extensions", ], + "@platforms//cpu:loongarch64": [ + "-mlsx", + "-mlasx", + ], "//conditions:default": [ ], }) @@ -193,6 +197,10 @@ cc_library( "//conditions:default": [], }) + select({ "@platforms//cpu:riscv64": ["hwy/ops/rvv-inl.h"], + "@platforms//cpu:loongarch64": [ + "hwy/ops/loongarch_lsx-inl.h", + "hwy/ops/loongarch_lasx-inl.h", + ], "//conditions:default": [], }), ) @@ -580,6 +588,7 @@ HWY_TEST_DEPS = [ # Fixes OOM for matvec_test on RVV. exec_properties = select({ "@platforms//cpu:riscv64": {"mem": "16g"}, + "@platforms//cpu:loongarch64": {"mem": "16g"}, "//conditions:default": None, }), features = select({ @@ -599,6 +608,7 @@ HWY_TEST_DEPS = [ }), linkstatic = select({ "@platforms//cpu:riscv64": True, + "@platforms//cpu:loongarch64": True, "//conditions:default": False, }), local_defines = ["HWY_IS_TEST"], diff --git a/WORKSPACE b/WORKSPACE index 8dab288d06..78c9d89aa5 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -4,13 +4,13 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive") load("@bazel_tools//tools/build_defs/repo:utils.bzl", "maybe") # Bazel platform rules. -maybe( - http_archive, +http_archive( name = "platforms", urls = [ - "https://github.com/bazelbuild/platforms/releases/download/0.0.7/platforms-0.0.7.tar.gz", + "https://mirror.bazel.build/github.com/bazelbuild/platforms/releases/download/1.0.0/platforms-1.0.0.tar.gz", + "https://github.com/bazelbuild/platforms/releases/download/1.0.0/platforms-1.0.0.tar.gz", ], - sha256 = "3a561c99e7bdbe9173aa653fd579fe849f1d8d67395780ab4770b1f381431d51", + sha256 = "3384eb1c30762704fbe38e440204e114154086c8fc8a8c2e3e28441028c019a8", ) maybe(