unsigned int            divisor;
        struct idr              handle_idr;
        struct rcu_head         rcu;
+       u32                     flags;
        /* The 'ht' field MUST be the last field in structure to allow for
         * more entries allocated at end of structure.
         */
        struct tcf_block *block = tp->chain->block;
        struct tc_cls_u32_offload cls_u32 = {};
 
-       tc_cls_common_offload_init_deprecated(&cls_u32.common, tp, NULL);
+       tc_cls_common_offload_init(&cls_u32.common, tp, h->flags, NULL);
        cls_u32.command = TC_CLSU32_DELETE_HNODE;
        cls_u32.hnode.divisor = h->divisor;
        cls_u32.hnode.handle = h->handle;
        bool offloaded = false;
        int err;
 
-       tc_cls_common_offload_init_deprecated(&cls_u32.common, tp, extack);
+       tc_cls_common_offload_init(&cls_u32.common, tp, flags, extack);
        cls_u32.command = TC_CLSU32_NEW_HNODE;
        cls_u32.hnode.divisor = h->divisor;
        cls_u32.hnode.handle = h->handle;
        struct tcf_block *block = tp->chain->block;
        struct tc_cls_u32_offload cls_u32 = {};
 
-       tc_cls_common_offload_init_deprecated(&cls_u32.common, tp, NULL);
+       tc_cls_common_offload_init(&cls_u32.common, tp, n->flags, NULL);
        cls_u32.command = TC_CLSU32_DELETE_KNODE;
        cls_u32.knode.handle = n->handle;
 
        bool skip_sw = tc_skip_sw(flags);
        int err;
 
-       tc_cls_common_offload_init_deprecated(&cls_u32.common, tp, extack);
+       tc_cls_common_offload_init(&cls_u32.common, tp, flags, extack);
        cls_u32.command = TC_CLSU32_REPLACE_KNODE;
        cls_u32.knode.handle = n->handle;
        cls_u32.knode.fshift = n->fshift;
                ht->handle = handle;
                ht->prio = tp->prio;
                idr_init(&ht->handle_idr);
+               ht->flags = flags;
 
                err = u32_replace_hw_hnode(tp, ht, flags, extack);
                if (err) {