On Wed, Jun 22, 2022 at 11:16:19PM +1200, Kai Huang wrote: >@@ -55,6 +55,7 @@ > #include <asm/uv/uv.h> > #include <asm/setup.h> > #include <asm/ftrace.h> >+#include <asm/tdx.h> > > #include "mm_internal.h" > >@@ -972,6 +973,26 @@ int arch_add_memory(int nid, u64 start, u64 size, > return add_pages(nid, start_pfn, nr_pages, params); > } > >+int arch_memory_add_precheck(int nid, u64 start, u64 size, mhp_t mhp_flags) >+{ >+ if (!platform_tdx_enabled()) >+ return 0; add a new cc attribute (if existing ones don't fit) for TDX host platform and check the attribute here. So that the code here can be reused by other cc platforms if they have the same requirement.