CUTLASS 2.10 (#615)

Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2022-09-03 18:48:46 -04:00
parent ca23ff7924
commit b72cbf957d
289 changed files with 43708 additions and 2513 deletions
@@ -0,0 +1,9 @@
+import pycutlass
+import unittest
+
+if __name__ == '__main__':
+    pycutlass.get_memory_pool(2**26, 2**26)
+    loader = unittest.TestLoader()
+    tests = loader.discover('./', 'gemm_*.py')
+    testRunner = unittest.runner.TextTestRunner()
+    testRunner.run(tests)