diff --git a/.bazelrc b/.bazelrc index e7efd3abdf5..ce8406b58aa 100644 --- a/.bazelrc +++ b/.bazelrc @@ -107,6 +107,9 @@ build --per_file_copt='//:aten/src/ATen/RegisterNestedTensorCPU\.cpp$'@-Wno-erro build --per_file_copt='//:aten/src/ATen/RegisterQuantizedCPU\.cpp$'@-Wno-error=unused-function build --per_file_copt='//:aten/src/ATen/RegisterSparseCPU\.cpp$'@-Wno-error=unused-function build --per_file_copt='//:aten/src/ATen/RegisterSparseCsrCPU\.cpp$'@-Wno-error=unused-function +build --per_file_copt='//:aten/src/ATen/RegisterNestedTensorMeta\.cpp$'@-Wno-error=unused-function +build --per_file_copt='//:aten/src/ATen/RegisterSparseMeta\.cpp$'@-Wno-error=unused-function +build --per_file_copt='//:aten/src/ATen/RegisterQuantizedMeta\.cpp$'@-Wno-error=unused-function build --per_file_copt='//:aten/src/ATen/RegisterZeroTensor\.cpp$'@-Wno-error=unused-function build --per_file_copt='//:torch/csrc/lazy/generated/RegisterAutogradLazy\.cpp$'@-Wno-error=unused-function build --per_file_copt='//:torch/csrc/lazy/generated/RegisterLazy\.cpp$'@-Wno-error=unused-function diff --git a/BUILD.bazel b/BUILD.bazel index 98aad88de06..9c6be5cc45f 100644 --- a/BUILD.bazel +++ b/BUILD.bazel @@ -50,6 +50,9 @@ generated_cpu_cpp = [ "aten/src/ATen/RegisterCompositeExplicitAutograd.cpp", "aten/src/ATen/RegisterCompositeExplicitAutogradNonFunctional.cpp", "aten/src/ATen/RegisterMeta.cpp", + "aten/src/ATen/RegisterSparseMeta.cpp", + "aten/src/ATen/RegisterQuantizedMeta.cpp", + "aten/src/ATen/RegisterNestedTensorMeta.cpp", "aten/src/ATen/RegisterSchema.cpp", "aten/src/ATen/CPUFunctions.h", "aten/src/ATen/CPUFunctions_inl.h", diff --git a/build.bzl b/build.bzl index afb8bd5e76c..016a91475c1 100644 --- a/build.bzl +++ b/build.bzl @@ -193,6 +193,9 @@ GENERATED_CPP = [ "RegisterCompositeImplicitAutograd.cpp", "RegisterZeroTensor.cpp", "RegisterMeta.cpp", + "RegisterQuantizedMeta.cpp", + "RegisterNestedTensorMeta.cpp", + "RegisterSparseMeta.cpp", "RegisterCompositeExplicitAutograd.cpp", "RegisterCompositeExplicitAutogradNonFunctional.cpp", "CompositeViewCopyKernels.cpp", diff --git a/torchgen/model.py b/torchgen/model.py index a247685e70a..de0926793c9 100644 --- a/torchgen/model.py +++ b/torchgen/model.py @@ -218,6 +218,9 @@ dispatch_keys = [ # Meta is a magic key: it is automatically generated for structured # kernels DispatchKey.Meta, + DispatchKey.SparseMeta, + DispatchKey.QuantizedMeta, + DispatchKey.NestedTensorMeta, DispatchKey.ZeroTensor, ]