actually run example with vector extensions

Author: kaushikcfd

test/test_target.py

@@ -708,7 +708,7 @@ def test_vec_tag_with_omp_simd_fallback():
         """,
         [lp.GlobalArg("x, y", shape=lp.auto, dtype=float)],
         seq_dependencies=True,
-        target=lp.CVectorExtensionsTarget())
+        target=lp.ExecutableCVectorExtensionsTarget())
 
     knl = lp.tag_inames(knl, {"j1": lp.VectorizeTag(lp.OpenMPSIMDTag()),
                               "j2": lp.VectorizeTag(lp.OpenMPSIMDTag()),
@@ -721,6 +721,10 @@ def test_vec_tag_with_omp_simd_fallback():
                 for line in code_str.split("\n")
                 if line.strip() == "#pragma omp simd"]) == 2
 
+    x = np.random.rand(10, 4)
+    _, (out,) = knl(x=x)
+    np.testing.assert_allclose(out, 2*x+1)
+
 
 def test_vec_extensions_with_multiple_loopy_body_insns():
     knl = lp.make_kernel(
@@ -734,7 +738,7 @@ def test_vec_extensions_with_multiple_loopy_body_insns():
         end
         """,
         seq_dependencies=True,
-        target=lp.CVectorExtensionsTarget())
+        target=lp.ExecutableCVectorExtensionsTarget())
 
     knl = lp.add_dtypes(knl, {"dat0": "float64"})
     knl = lp.split_iname(knl, "n", 4, slabs=(1, 1),
@@ -744,7 +748,8 @@ def test_vec_extensions_with_multiple_loopy_body_insns():
     knl = lp.tag_inames(knl, {
         "n_batch": lp.VectorizeTag(lp.OpenMPSIMDTag())})
 
-    print(lp.generate_code_v2(knl).device_code())
+    _, (out,) = knl(N=100)
+    np.testing.assert_allclose(out, 2*np.ones((100, 1)))
 
 
 if __name__ == "__main__":