mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
41690d7804
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70861 ghstack-source-id: 147642549 Test Plan: Should be a no-op. Rely on CI to validate. Reviewed By: malfet Differential Revision: D33329870 fbshipit-source-id: 7dbccaa994737c5fe7195d02dffd61eeceb19ceb (cherry picked from commit 2b5264ebc49e4a5445c066e07f15bad041f42ac8)
33 lines
900 B
Python
33 lines
900 B
Python
def define_targets(rules):
|
|
rules.cc_library(
|
|
name = "CPUCachingAllocator",
|
|
srcs = ["CPUCachingAllocator.cpp"],
|
|
hdrs = ["CPUCachingAllocator.h"],
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"//c10/core:alloc_cpu",
|
|
"//c10/util:base",
|
|
],
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "CPUProfilingAllocator",
|
|
srcs = ["CPUProfilingAllocator.cpp"],
|
|
hdrs = ["CPUProfilingAllocator.h"],
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"//c10/core:alloc_cpu",
|
|
"//c10/util:base",
|
|
],
|
|
)
|
|
|
|
rules.filegroup(
|
|
name = "headers",
|
|
srcs = rules.glob(["*.h"]),
|
|
visibility = ["//c10:__pkg__"],
|
|
)
|