Added test case for cuda_kernel and cuda_device callee types in callee_type filter (#263)

src/config/config.cc

@@ -106,6 +106,10 @@ std::string to_string(callee_type mt)
         return "function_template";
     case callee_type::lambda:
         return "lambda";
+    case callee_type::cuda_kernel:
+        return "cuda_kernel";
+    case callee_type::cuda_device:
+        return "cuda_device";
     }
 
     assert(false);

src/config/config.h

@@ -78,7 +78,9 @@ enum class callee_type {
     method,
     function,
     function_template,
-    lambda
+    lambda,
+    cuda_kernel,
+    cuda_device
 };
 
 std::string to_string(callee_type mt);

src/config/schema.h

@@ -112,6 +112,8 @@ types:
         - function
         - function_template
         - lambda
+        - cuda_kernel
+        - cuda_device
     context_filter_match_t:
         match:
             radius: int

src/config/yaml_decoders.cc

@@ -312,6 +312,10 @@ template <> struct convert<callee_type> {
             rhs = callee_type::method;
         else if (val == to_string(callee_type::lambda))
             rhs = callee_type::lambda;
+        else if (val == to_string(callee_type::cuda_kernel))
+            rhs = callee_type::cuda_kernel;
+        else if (val == to_string(callee_type::cuda_device))
+            rhs = callee_type::cuda_device;
         else
             return false;