[BugFix] Correct argparse for example_convolution test (#474)

* add convolution example to CI

* lint fix

* Update test_example_convolution.py

* fix bug

---------
Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>

examples/convolution/example_convolution.py

@@ -230,7 +230,7 @@ def convolution(N,
     return main
 
 
-def main():
+def main(argv=None):
     parser = argparse.ArgumentParser()
     parser.add_argument('--n', type=int, default=128, help='n')
     parser.add_argument('--c', type=int, default=128, help='c')
@@ -252,7 +252,7 @@ def main():
         default=True,
         help="Whether to enable BitBLAS roller for search space")
 
-    args = parser.parse_args()
+    args = parser.parse_args(argv)
     N, C, H, W, F, K, S, D, P = args.n, args.c, args.h, args.w, args.f, args.k, args.s, args.d, args.p
     a = torch.randn(N, H, W, C).cuda().half()
     b = torch.randn(K, K, C, F).cuda().half()

examples/convolution/test_example_convolution.py

@@ -5,7 +5,7 @@ import example_convolution
 
 @tilelang.testing.requires_cuda
 def test_example_convolution():
-    example_convolution.main()
+    example_convolution.main([])
 
 
 if __name__ == "__main__":