size_t                  mem_size;
        size_t                  num_desc;
        struct gen_pool         *gen_pool;
+       void                    **desc_infos;
 };
 
 void k3_cppi_desc_pool_destroy(struct k3_cppi_desc_pool *pool)
                dma_free_coherent(pool->dev, pool->mem_size, pool->cpumem,
                                  pool->dma_addr);
 
+       kfree(pool->desc_infos);
+
        gen_pool_destroy(pool->gen_pool);       /* frees pool->name */
+
+       kfree(pool);
 }
 EXPORT_SYMBOL_GPL(k3_cppi_desc_pool_destroy);
 
        const char *pool_name = NULL;
        int ret = -ENOMEM;
 
-       pool = devm_kzalloc(dev, sizeof(*pool), GFP_KERNEL);
+       pool = kzalloc(sizeof(*pool), GFP_KERNEL);
        if (!pool)
                return ERR_PTR(ret);
 
        pool_name = kstrdup_const(name ? name : dev_name(pool->dev),
                                  GFP_KERNEL);
        if (!pool_name)
-               return ERR_PTR(-ENOMEM);
+               goto gen_pool_create_fail;
 
        pool->gen_pool = gen_pool_create(ilog2(pool->desc_size), -1);
        if (!pool->gen_pool) {
-               ret = -ENOMEM;
-               dev_err(pool->dev, "pool create failed %d\n", ret);
                kfree_const(pool_name);
                goto gen_pool_create_fail;
        }
 
        pool->gen_pool->name = pool_name;
 
+       pool->desc_infos = kcalloc(pool->num_desc,
+                                  sizeof(*pool->desc_infos), GFP_KERNEL);
+       if (!pool->desc_infos)
+               goto gen_pool_desc_infos_alloc_fail;
+
        pool->cpumem = dma_alloc_coherent(pool->dev, pool->mem_size,
                                          &pool->dma_addr, GFP_KERNEL);
 
        dma_free_coherent(pool->dev, pool->mem_size, pool->cpumem,
                          pool->dma_addr);
 dma_alloc_fail:
+       kfree(pool->desc_infos);
+gen_pool_desc_infos_alloc_fail:
        gen_pool_destroy(pool->gen_pool);       /* frees pool->name */
 gen_pool_create_fail:
-       devm_kfree(pool->dev, pool);
+       kfree(pool);
        return ERR_PTR(ret);
 }
 EXPORT_SYMBOL_GPL(k3_cppi_desc_pool_create_name);
 }
 EXPORT_SYMBOL_GPL(k3_cppi_desc_pool_cpuaddr);
 
+void k3_cppi_desc_pool_desc_info_set(struct k3_cppi_desc_pool *pool,
+                                    int desc_idx, void *info)
+{
+       pool->desc_infos[desc_idx] = info;
+}
+EXPORT_SYMBOL_GPL(k3_cppi_desc_pool_desc_info_set);
+
+void *k3_cppi_desc_pool_desc_info(const struct k3_cppi_desc_pool *pool,
+                                 int desc_idx)
+{
+       return pool->desc_infos[desc_idx];
+}
+EXPORT_SYMBOL_GPL(k3_cppi_desc_pool_desc_info);
+
 MODULE_LICENSE("GPL");
 MODULE_DESCRIPTION("TI K3 CPPI5 descriptors pool API");