1 files changed, 2 insertions, 4 deletions

diff --git a/drivers/infiniband/hw/mlx4/main.c b/drivers/infiniband/hw/mlx4/main.c
index 0f7027e7db13..0dc807f7b069 100644
--- a/drivers/infiniband/hw/mlx4/main.c
+++ b/drivers/infiniband/hw/mlx4/main.c
@@ -910,8 +910,7 @@ static int __mlx4_ib_default_rules_match(struct ib_qp *qp,
         const struct default_rules *pdefault_rules = default_table;
         u8 link_layer = rdma_port_get_link_layer(qp->device, flow_attr->port);
 
-        for (i = 0; i < sizeof(default_table)/sizeof(default_table[0]); i++,
-             pdefault_rules++) {
+        for (i = 0; i < ARRAY_SIZE(default_table); i++, pdefault_rules++) {
                 __u32 field_types[IB_FLOW_SPEC_SUPPORT_LAYERS];
                 memset(&field_types, 0, sizeof(field_types));
 
@@ -965,8 +964,7 @@ static int __mlx4_ib_create_default_rules(
         int size = 0;
         int i;
 
-        for (i = 0; i < sizeof(pdefault_rules->rules_create_list)/
-                        sizeof(pdefault_rules->rules_create_list[0]); i++) {
+        for (i = 0; i < ARRAY_SIZE(pdefault_rules->rules_create_list); i++) {
                 int ret;
                 union ib_flow_spec ib_spec;
                 switch (pdefault_rules->rules_create_list[i]) {