Skip to content

Commit e1e7cd1

Browse files
authored
Blaze aios dev hipblaslt hsaco cache merge (#37) (#2688)
* replace memory cache by file cache * add memory cache * fix mcpu * insert cache in the correct place. * remove duplicate bc * fix bugs * add compile hsaco unit test * fix hsaco name by hash ir * fix unit test
1 parent 4483ada commit e1e7cd1

File tree

4 files changed

+202
-67
lines changed

4 files changed

+202
-67
lines changed

tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/BUILD

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,3 +72,23 @@ tf_cc_test(
7272
"@llvm//:support",
7373
],
7474
)
75+
76+
77+
tf_cc_test(
78+
name = "gpu_backend_lib_amdgpu_test",
79+
size = "small",
80+
srcs = ["gpu_backend_lib_test.cc"],
81+
deps = [
82+
":llvm_gpu_backend",
83+
"//tensorflow/compiler/xla:statusor",
84+
"//tensorflow/compiler/xla/service:hlo_module_config",
85+
"//tensorflow/compiler/xla/service/gpu:gpu_types",
86+
"//tensorflow/compiler/xla/service/gpu:target_constants",
87+
"//tensorflow/core:test",
88+
"//tensorflow/core:test_main",
89+
"//tensorflow/core/platform:rocm_rocdl_path",
90+
"@llvm//:core",
91+
"@llvm//:support",
92+
"@com_google_protobuf//:protobuf",
93+
],
94+
)

0 commit comments

Comments
 (0)