(oct, CN23XX_SLI_PKT_MAC_RINFO64(mac_no, pf_num)));
 }
 
+static int cn23xx_reset_io_queues(struct octeon_device *oct)
+{
+       int ret_val = 0;
+       u64 d64;
+       u32 q_no, srn, ern;
+       u32 loop = 1000;
+
+       srn = oct->sriov_info.pf_srn;
+       ern = srn + oct->sriov_info.num_pf_rings;
+
+       /*As per HRM reg description, s/w cant write 0 to ENB. */
+       /*to make the queue off, need to set the RST bit. */
+
+       /* Reset the Enable bit for all the 64 IQs.  */
+       for (q_no = srn; q_no < ern; q_no++) {
+               /* set RST bit to 1. This bit applies to both IQ and OQ */
+               d64 = octeon_read_csr64(oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no));
+               d64 = d64 | CN23XX_PKT_INPUT_CTL_RST;
+               octeon_write_csr64(oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no), d64);
+       }
+
+       /*wait until the RST bit is clear or the RST and quite bits are set*/
+       for (q_no = srn; q_no < ern; q_no++) {
+               u64 reg_val = octeon_read_csr64(oct,
+                                       CN23XX_SLI_IQ_PKT_CONTROL64(q_no));
+               while ((READ_ONCE(reg_val) & CN23XX_PKT_INPUT_CTL_RST) &&
+                      !(READ_ONCE(reg_val) & CN23XX_PKT_INPUT_CTL_QUIET) &&
+                      loop--) {
+                       WRITE_ONCE(reg_val, octeon_read_csr64(
+                           oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no)));
+               }
+               if (!loop) {
+                       dev_err(&oct->pci_dev->dev,
+                               "clearing the reset reg failed or setting the quiet reg failed for qno: %u\n",
+                               q_no);
+                       return -1;
+               }
+               WRITE_ONCE(reg_val, READ_ONCE(reg_val) &
+                       ~CN23XX_PKT_INPUT_CTL_RST);
+               octeon_write_csr64(oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no),
+                                  READ_ONCE(reg_val));
+
+               WRITE_ONCE(reg_val, octeon_read_csr64(
+                          oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no)));
+               if (READ_ONCE(reg_val) & CN23XX_PKT_INPUT_CTL_RST) {
+                       dev_err(&oct->pci_dev->dev,
+                               "clearing the reset failed for qno: %u\n",
+                               q_no);
+                       ret_val = -1;
+               }
+       }
+
+       return ret_val;
+}
+
 static int cn23xx_pf_setup_global_input_regs(struct octeon_device *oct)
 {
        u32 q_no, ern, srn;
        srn = oct->sriov_info.pf_srn;
        ern = srn + oct->sriov_info.num_pf_rings;
 
+       if (cn23xx_reset_io_queues(oct))
+               return -1;
+
        /** Set the MAC_NUM and PVF_NUM in IQ_PKT_CONTROL reg
        * for all queues.Only PF can set these bits.
        * bits 29:30 indicate the MAC num.
                         reg_val);
 }
 
+static int cn23xx_enable_io_queues(struct octeon_device *oct)
+{
+       u64 reg_val;
+       u32 srn, ern, q_no;
+       u32 loop = 1000;
+
+       srn = oct->sriov_info.pf_srn;
+       ern = srn + oct->num_iqs;
+
+       for (q_no = srn; q_no < ern; q_no++) {
+               /* set the corresponding IQ IS_64B bit */
+               if (oct->io_qmask.iq64B & BIT_ULL(q_no - srn)) {
+                       reg_val = octeon_read_csr64(
+                           oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no));
+                       reg_val = reg_val | CN23XX_PKT_INPUT_CTL_IS_64B;
+                       octeon_write_csr64(
+                           oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no), reg_val);
+               }
+
+               /* set the corresponding IQ ENB bit */
+               if (oct->io_qmask.iq & BIT_ULL(q_no - srn)) {
+                       /* IOQs are in reset by default in PEM2 mode,
+                        * clearing reset bit
+                        */
+                       reg_val = octeon_read_csr64(
+                           oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no));
+
+                       if (reg_val & CN23XX_PKT_INPUT_CTL_RST) {
+                               while ((reg_val & CN23XX_PKT_INPUT_CTL_RST) &&
+                                      !(reg_val &
+                                        CN23XX_PKT_INPUT_CTL_QUIET) &&
+                                      loop--) {
+                                       reg_val = octeon_read_csr64(
+                                           oct,
+                                           CN23XX_SLI_IQ_PKT_CONTROL64(q_no));
+                               }
+                               if (!loop) {
+                                       dev_err(&oct->pci_dev->dev,
+                                               "clearing the reset reg failed or setting the quiet reg failed for qno: %u\n",
+                                               q_no);
+                                       return -1;
+                               }
+                               reg_val = reg_val & ~CN23XX_PKT_INPUT_CTL_RST;
+                               octeon_write_csr64(
+                                   oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no),
+                                   reg_val);
+
+                               reg_val = octeon_read_csr64(
+                                   oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no));
+                               if (reg_val & CN23XX_PKT_INPUT_CTL_RST) {
+                                       dev_err(&oct->pci_dev->dev,
+                                               "clearing the reset failed for qno: %u\n",
+                                               q_no);
+                                       return -1;
+                               }
+                       }
+                       reg_val = octeon_read_csr64(
+                           oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no));
+                       reg_val = reg_val | CN23XX_PKT_INPUT_CTL_RING_ENB;
+                       octeon_write_csr64(
+                           oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no), reg_val);
+               }
+       }
+       for (q_no = srn; q_no < ern; q_no++) {
+               u32 reg_val;
+               /* set the corresponding OQ ENB bit */
+               if (oct->io_qmask.oq & BIT_ULL(q_no - srn)) {
+                       reg_val = octeon_read_csr(
+                           oct, CN23XX_SLI_OQ_PKT_CONTROL(q_no));
+                       reg_val = reg_val | CN23XX_PKT_OUTPUT_CTL_RING_ENB;
+                       octeon_write_csr(oct, CN23XX_SLI_OQ_PKT_CONTROL(q_no),
+                                        reg_val);
+               }
+       }
+       return 0;
+}
+
+static void cn23xx_disable_io_queues(struct octeon_device *oct)
+{
+       int q_no, loop;
+       u64 d64;
+       u32 d32;
+       u32 srn, ern;
+
+       srn = oct->sriov_info.pf_srn;
+       ern = srn + oct->num_iqs;
+
+       /*** Disable Input Queues. ***/
+       for (q_no = srn; q_no < ern; q_no++) {
+               loop = HZ;
+
+               /* start the Reset for a particular ring */
+               WRITE_ONCE(d64, octeon_read_csr64(
+                          oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no)));
+               WRITE_ONCE(d64, READ_ONCE(d64) &
+                                       (~(CN23XX_PKT_INPUT_CTL_RING_ENB)));
+               WRITE_ONCE(d64, READ_ONCE(d64) | CN23XX_PKT_INPUT_CTL_RST);
+               octeon_write_csr64(oct, CN23XX_SLI_IQ_PKT_CONTROL64(q_no),
+                                  READ_ONCE(d64));
+
+               /* Wait until hardware indicates that the particular IQ
+                * is out of reset.
+                */
+               WRITE_ONCE(d64, octeon_read_csr64(
+                                       oct, CN23XX_SLI_PKT_IOQ_RING_RST));
+               while (!(READ_ONCE(d64) & BIT_ULL(q_no)) && loop--) {
+                       WRITE_ONCE(d64, octeon_read_csr64(
+                                       oct, CN23XX_SLI_PKT_IOQ_RING_RST));
+                       schedule_timeout_uninterruptible(1);
+               }
+
+               /* Reset the doorbell register for this Input Queue. */
+               octeon_write_csr(oct, CN23XX_SLI_IQ_DOORBELL(q_no), 0xFFFFFFFF);
+               while (octeon_read_csr64(oct, CN23XX_SLI_IQ_DOORBELL(q_no)) &&
+                      loop--) {
+                       schedule_timeout_uninterruptible(1);
+               }
+       }
+
+       /*** Disable Output Queues. ***/
+       for (q_no = srn; q_no < ern; q_no++) {
+               loop = HZ;
+
+               /* Wait until hardware indicates that the particular IQ
+                * is out of reset.It given that SLI_PKT_RING_RST is
+                * common for both IQs and OQs
+                */
+               WRITE_ONCE(d64, octeon_read_csr64(
+                                       oct, CN23XX_SLI_PKT_IOQ_RING_RST));
+               while (!(READ_ONCE(d64) & BIT_ULL(q_no)) && loop--) {
+                       WRITE_ONCE(d64, octeon_read_csr64(
+                                       oct, CN23XX_SLI_PKT_IOQ_RING_RST));
+                       schedule_timeout_uninterruptible(1);
+               }
+
+               /* Reset the doorbell register for this Output Queue. */
+               octeon_write_csr(oct, CN23XX_SLI_OQ_PKTS_CREDIT(q_no),
+                                0xFFFFFFFF);
+               while (octeon_read_csr64(oct,
+                                        CN23XX_SLI_OQ_PKTS_CREDIT(q_no)) &&
+                      loop--) {
+                       schedule_timeout_uninterruptible(1);
+               }
+
+               /* clear the SLI_PKT(0..63)_CNTS[CNT] reg value */
+               WRITE_ONCE(d32, octeon_read_csr(
+                                       oct, CN23XX_SLI_OQ_PKTS_SENT(q_no)));
+               octeon_write_csr(oct, CN23XX_SLI_OQ_PKTS_SENT(q_no),
+                                READ_ONCE(d32));
+       }
+}
+
 static void cn23xx_get_pcie_qlmport(struct octeon_device *oct)
 {
        oct->pcie_port = (octeon_read_csr(oct, CN23XX_SLI_MAC_NUMBER)) & 0xff;
        oct->fn_list.setup_oq_regs = cn23xx_setup_oq_regs;
        oct->fn_list.setup_device_regs = cn23xx_setup_pf_device_regs;
 
+       oct->fn_list.enable_io_queues = cn23xx_enable_io_queues;
+       oct->fn_list.disable_io_queues = cn23xx_disable_io_queues;
+
        cn23xx_setup_reg_address(oct);
 
        oct->coproc_clock_rate = 1000000ULL * cn23xx_coprocessor_clock(oct);