wehub-resource-sync
71 lines
1.4 KiB
Python
71 lines
1.4 KiB
Python
# StableHLO Device Profile reference implementation
|
|
|
|
load("@rules_cc//cc:cc_library.bzl", "cc_library")
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
|
|
default_visibility = [":__subpackages__"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "shlo",
|
|
srcs = [
|
|
"src/broadcast_in_dim.cc",
|
|
"src/clamp.cc",
|
|
"src/compare.cc",
|
|
"src/concatenate.cc",
|
|
"src/dispatch.h",
|
|
"src/elementwise_binary.cc",
|
|
"src/elementwise_unary.cc",
|
|
"src/iota.cc",
|
|
"src/is_finite.cc",
|
|
"src/select.cc",
|
|
"src/shlo.cc",
|
|
"src/uniform_dequantize_quantize.cc",
|
|
],
|
|
hdrs = [
|
|
"include/shlo.h",
|
|
"src/storage.h",
|
|
"src/util.h",
|
|
],
|
|
tags = ["no_oss"],
|
|
deps = [
|
|
":float",
|
|
"@com_google_absl//absl/log",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_absl//absl/types:span",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "debug",
|
|
srcs = [
|
|
"src/debug.cc",
|
|
],
|
|
hdrs = [
|
|
"src/debug.h",
|
|
],
|
|
tags = ["no_oss"],
|
|
deps = [
|
|
":float",
|
|
":shlo",
|
|
"@com_google_absl//absl/log",
|
|
"@com_google_absl//absl/log:check",
|
|
"@com_google_absl//absl/types:span",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "float",
|
|
srcs = [
|
|
],
|
|
hdrs = [
|
|
"src/bf16.h",
|
|
"src/f16.h",
|
|
"src/has_keyword.h",
|
|
],
|
|
tags = ["no_oss"],
|
|
deps = [
|
|
],
|
|
)
|