Aleksandr Pivovar 4a68cf748e
added support of b2b bmm ( #849 ) ...
* added support of b2b bmm
* fixed arguments and params structures
* added batch_count argument
* removed SplitKSerial and added new test case with b2b bmm
* fixed support of Kbatched and added new test case with batch stride
* added batch support for bias and scale
* make test
* small changes
---------
Co-authored-by: Haicheng Wu <haichengw@nvidia.com >
2023-04-14 23:20:02 -04:00
2023-03-09 23:22:56 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-03-09 23:22:56 -05:00
2023-03-09 23:22:56 -05:00
2023-04-14 23:19:34 -04:00
2023-01-23 20:55:28 -05:00
2023-01-23 20:55:28 -05:00
2023-01-20 16:32:57 -05:00
2023-04-14 23:20:02 -04:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-04-14 23:19:34 -04:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-04-14 23:19:34 -04:00
2023-03-09 23:22:56 -05:00
2023-04-14 23:19:34 -04:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-04-14 23:19:34 -04:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-04-03 10:01:12 -04:00
2023-01-20 16:32:57 -05:00
2023-01-23 20:55:28 -05:00
2023-01-20 16:32:57 -05:00
2023-04-14 23:19:34 -04:00
2023-04-14 23:19:34 -04:00
2023-04-06 20:44:58 -04:00
2023-01-20 16:32:57 -05:00
2023-01-20 16:32:57 -05:00
2023-03-09 23:22:56 -05:00
2023-04-14 23:19:34 -04:00
2023-04-14 23:19:34 -04:00
2023-01-23 20:55:28 -05:00
2023-04-14 23:19:34 -04:00
2023-04-14 23:19:34 -04:00
2023-04-14 23:19:34 -04:00
2023-04-14 23:19:34 -04:00
2023-01-20 16:32:57 -05:00
2023-04-14 23:19:34 -04:00
2023-04-14 23:19:34 -04:00
2023-04-14 23:19:34 -04:00
2023-04-14 23:19:34 -04:00
4a68cf748e