diff --git a/sycl/test/format.py b/sycl/test/format.py
index 19d16a263426c..eb3a497ecd004 100644
--- a/sycl/test/format.py
+++ b/sycl/test/format.py
@@ -5,6 +5,8 @@
 import os
 import re
 
+SUFFIXES = {".hpp"}
+
 
 class SYCLHeadersTest(lit.formats.TestFormat):
     def getTestsForPath(
@@ -21,7 +23,8 @@ def getTestsInDirectory(self, testSuite, path_in_suite, litConfig, localConfig):
         for dirpath, _, filenames in os.walk(source_path):
             relative_dirpath = dirpath[len(localConfig.sycl_include) + 1 :]
             for filename in filenames:
-                if not filename.endswith(".hpp"):
+                suffix = os.path.splitext(filename)[1]
+                if suffix not in SUFFIXES or suffix not in litConfig.suffixes:
                     continue
                 filepath = os.path.join(dirpath, filename)