[experimental][kleidi] rebase fixes with int to size_t

Author: digantdesai

torchao/experimental/kernels/cpu/aarch64/kleidi/kai_matmul_clamp_f32_qai8dxp1x8_qsi4c32p4x8_1x4x32_neon_dotprod.h

@@ -40,7 +40,7 @@ const Ukernel get_ukernel() {
           kai_run_matmul_clamp_f32_qai8dxp1x8_qsi4c32p4x8_1x4x32_neon_dotprod};
 }
 
-int activation_data_size(int m, int k, int group_size) {
+size_t activation_data_size(int m, int k, int group_size) {
   (void)group_size; // unused
   return kai_matmul_clamp_f32_qai8dxp_qsi4c32p::activation_data_size(
       get_ukernel(), m, k);
@@ -57,7 +57,7 @@ void prepare_activation_data(
       get_ukernel(), activation_data, m, k, activations);
 }
 
-int weight_data_size(int n, int k, int group_size) {
+size_t weight_data_size(int n, int k, int group_size) {
   return kai_matmul_clamp_f32_qai8dxp_qsi4c32p::weight_data_size(
       get_ukernel(), n, k, group_size);
 }
@@ -115,7 +115,7 @@ void kernel(
       clamp_max);
 }
 
-size_t get_alignement() {
+size_t get_preferred_alignement() {
   return 16;
 }
 } // namespace neon_dotprod_1x4x32

torchao/experimental/kernels/cpu/aarch64/kleidi/kai_matmul_clamp_f32_qai8dxp1x8_qsi4c32p8x8_1x8x32_neon_dotprod.h

@@ -39,7 +39,7 @@ const Ukernel get_ukernel() {
           kai_run_matmul_clamp_f32_qai8dxp1x8_qsi4c32p8x8_1x8x32_neon_dotprod};
 }
 
-int activation_data_size(int m, int k, int group_size) {
+size_t activation_data_size(int m, int k, int group_size) {
   (void) group_size; // unused
   return kai_matmul_clamp_f32_qai8dxp_qsi4c32p::activation_data_size(get_ukernel(), m, k);
 }
@@ -59,7 +59,7 @@ void prepare_activation_data(
       activations);
 }
 
-int weight_data_size(int n, int k, int group_size) {
+size_t weight_data_size(int n, int k, int group_size) {
   return kai_matmul_clamp_f32_qai8dxp_qsi4c32p::weight_data_size(get_ukernel(), n, k, group_size);
 }
 
@@ -116,7 +116,7 @@ void kernel(
       clamp_max);
 }
 
-size_t get_alignement() {
+size_t get_preferred_alignement() {
   return 16;
 }
 } // namespace neon_dotprod_1x4x32

torchao/experimental/kernels/cpu/aarch64/kleidi/kai_matmul_clamp_f32_qai8dxp_qsi4c32p.h

@@ -43,7 +43,7 @@ namespace kai_matmul_clamp_f32_qai8dxp_qsi4c32p {
 
 using Ukernel = struct kai_matmul_clamp_f32_qai8dxp_qsi4c32p_ukernel;
 
-int activation_data_size(const Ukernel ukernel, int m, int k) {
+size_t activation_data_size(const Ukernel ukernel, int m, int k) {
   auto lhs_packing = get_lhs_packing();
   return lhs_packing.get_lhs_packed_size(
       m, k, ukernel.get_mr(), ukernel.get_kr(), ukernel.get_sr());
@@ -69,7 +69,7 @@ void prepare_activation_data(
       activation_data);
 }
 
-int weight_data_size(const Ukernel ukernel, int n, int k, int group_size) {
+size_t weight_data_size(const Ukernel ukernel, int n, int k, int group_size) {
   auto rhs_pack = get_rhs_packing();
   return rhs_pack.get_rhs_packed_size(
       n,