1 files changed, 8 insertions, 0 deletions
diff --git a/src/mesa/drivers/dri/i965/brw_performance_query.c b/src/mesa/drivers/dri/i965/brw_performance_query.c
index 90a3d588bdf..efb0b9c3fab 100644
--- a/src/mesa/drivers/dri/i965/brw_performance_query.c
+++ b/src/mesa/drivers/dri/i965/brw_performance_query.c
@@ -78,6 +78,8 @@
 #include "brw_oa_sklgt3.h"
 #include "brw_oa_sklgt4.h"
 #include "brw_oa_bxt.h"
+#include "brw_oa_kblgt2.h"
+#include "brw_oa_kblgt3.h"
 #include "intel_batchbuffer.h"
 
 #define FILE_DEBUG_FLAG DEBUG_PERFMON
@@ -2009,6 +2011,12 @@ get_register_queries_function(const struct gen_device_info *devinfo)
       if (devinfo->gt == 4)
          return brw_oa_register_queries_sklgt4;
    }
+   if (devinfo->is_kabylake) {
+      if (devinfo->gt == 2)
+         return brw_oa_register_queries_kblgt2;
+      if (devinfo->gt == 3)
+         return brw_oa_register_queries_kblgt3;
+   }
    return NULL;
 }