diff --git a/tools/get_pr_ut.py b/tools/get_pr_ut.py
index 5dd9f71485e2580279071419c77609c8f2f6ea94..174f5d1756536bad854a223d125e17244a75e88f 100644
--- a/tools/get_pr_ut.py
+++ b/tools/get_pr_ut.py
@@ -188,7 +188,8 @@ class PRChecker(object):
                         ut_list.append('h_cu_comment_placeholder')
                     else:
                         return ''
-                elif f.endswith('.cc') or f.endswith('.py'):
+                elif f.endswith('.cc') or f.endswith('.py') or f.endswith(
+                        '.cu'):
                     if f.find('test_') != -1 or f.find('_test') != -1:
                         check_added_ut = True
                     elif self.is_only_comment(f):