diff --git a/util/cbmem/cbmem.c b/util/cbmem/cbmem.c
index 90ad55e070..8e73d9c28d 100644
--- a/util/cbmem/cbmem.c
+++ b/util/cbmem/cbmem.c
@@ -1123,7 +1123,7 @@ static void print_usage(const char *name, int exit_code)
 	exit(exit_code);
 }
 
-#ifdef __arm__
+#if defined(__arm__) || defined(__aarch64__)
 static void dt_update_cells(const char *name, int *addr_cells_ptr,
 			    int *size_cells_ptr)
 {
@@ -1236,7 +1236,7 @@ static char *dt_find_compat(const char *parent, const char *compat,
 	closedir(dir);
 	return ret;
 }
-#endif /* __arm__ */
+#endif /* defined(__arm__) || defined(__aarch64__) */
 
 int main(int argc, char** argv)
 {
@@ -1339,7 +1339,7 @@ int main(int argc, char** argv)
 		return 1;
 	}
 
-#ifdef __arm__
+#if defined(__arm__) || defined(__aarch64__)
 	int addr_cells, size_cells;
 	char *coreboot_node = dt_find_compat("/proc/device-tree", "coreboot",
 					     &addr_cells, &size_cells);