frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implement HPUHooksInterface (#137338)

Browse Source 
Fixes #137262

Pull Request resolved: https://github.com/pytorch/pytorch/pull/137338
Approved by: https://github.com/guangyey, https://github.com/albanD

Co-authored-by: Yu, Guangye <106960996+guangyey@users.noreply.github.com>
This commit is contained in:
Tomasz Bohutyn
2024-10-31 16:26:16 +00:00
committed by PyTorch MergeBot
parent 87f1990697
commit dd6263e2fb
5 changed files with 103 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
build_variables.bzl
View File

@ -974,6 +974,7 @@ def glob_libtorch_python_sources(gencode_pattern = ":generate-code[{}]"):
aten_cpu_non_globed_sources = [
"aten/src/ATen/detail/CUDAHooksInterface.cpp",
"aten/src/ATen/detail/HIPHooksInterface.cpp",
"aten/src/ATen/detail/HPUHooksInterface.cpp",
"aten/src/ATen/detail/MPSHooksInterface.cpp",
"aten/src/ATen/detail/MAIAHooksInterface.cpp",
"aten/src/ATen/detail/PrivateUse1HooksInterface.cpp",
@ -992,6 +993,7 @@ aten_cpu_non_globed_headers = [
"aten/src/ATen/detail/CUDAHooksInterface.h",
"aten/src/ATen/detail/MPSHooksInterface.h",
"aten/src/ATen/detail/HIPHooksInterface.h",
"aten/src/ATen/detail/HPUHooksInterface.h",
"aten/src/ATen/detail/MAIAHooksInterface.h",
"aten/src/ATen/detail/PrivateUse1HooksInterface.h",
"aten/src/ATen/detail/XPUHooksInterface.h",