From 701c8e06a072b4670eb3ab5ac85c0fc9e024a985 Mon Sep 17 00:00:00 2001
From: chalsliu <45041955+chalsliu@users.noreply.github.com>
Date: Wed, 9 Dec 2020 20:36:09 +0800
Subject: [PATCH] Support precision test for cuda new ut

---
 tools/get_pr_ut.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/get_pr_ut.py b/tools/get_pr_ut.py
index 5dd9f71485e..174f5d17565 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):
-- 
GitLab