thunderbolt: dma_test: Use enum tb_link_width

Since we have it, use it in the DMA test driver as well.

Signed-off-by: Mika Westerberg <mika.westerberg@linux.intel.com>

diff --git a/drivers/thunderbolt/dma_test.c b/drivers/thunderbolt/dma_test.c
index 39476fc488015b2470bc790e568308f8680a904d..9e47a63f28e786c297adacddab0137bf0c4fe6ce 100644 (file)
--- a/drivers/thunderbolt/dma_test.c
+++ b/drivers/thunderbolt/dma_test.c
@@ -101,7 +101,7 @@ struct dma_test {
        unsigned int packets_sent;
        unsigned int packets_received;
        unsigned int link_speed;
-       unsigned int link_width;
+       enum tb_link_width link_width;
        unsigned int crc_errors;
        unsigned int buffer_overflow_errors;
        enum dma_test_result result;
@@ -465,9 +465,9 @@ DMA_TEST_DEBUGFS_ATTR(packets_to_send, packets_to_send_get,
 static int dma_test_set_bonding(struct dma_test *dt)
 {
        switch (dt->link_width) {
-       case 2:
+       case TB_LINK_WIDTH_DUAL:
                return tb_xdomain_lane_bonding_enable(dt->xd);
-       case 1:
+       case TB_LINK_WIDTH_SINGLE:
                tb_xdomain_lane_bonding_disable(dt->xd);
                fallthrough;
        default:
@@ -490,12 +490,8 @@ static void dma_test_check_errors(struct dma_test *dt, int ret)
        if (!dt->error_code) {
                if (dt->link_speed && dt->xd->link_speed != dt->link_speed) {
                        dt->error_code = DMA_TEST_SPEED_ERROR;
-               } else if (dt->link_width) {
-                       const struct tb_xdomain *xd = dt->xd;
-
-                       if ((dt->link_width == 1 && xd->link_width != TB_LINK_WIDTH_SINGLE) ||
-                           (dt->link_width == 2 && xd->link_width < TB_LINK_WIDTH_DUAL))
-                               dt->error_code = DMA_TEST_WIDTH_ERROR;
+               } else if (dt->link_width && dt->link_width != dt->xd->link_width) {
+                       dt->error_code = DMA_TEST_WIDTH_ERROR;
                } else if (dt->packets_to_send != dt->packets_sent ||
                         dt->packets_to_receive != dt->packets_received ||
                         dt->crc_errors || dt->buffer_overflow_errors) {