Support precision test for cuda new ut

5 years ago · 701c8e06a0
parent 50d3117d30
commit 701c8e06a0
1 changed files with 2 additions and 1 deletions
--- 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):