Adam Osewski b4032629e5 Grouped GEMM Multiple D tile loop. (#1247) ...

* Overload output stream operator for LoopScheduler and PiplineVersion

* Add Run overload accepting grid descriptors MK.

* Add __device__ keyword for CalculateGridSize

* Create device op GroupedGemmMultipleD

* Add GroupedGemm MultipleD Tile Loop implementation.

* Add an example for GroupedGemm MultipleD tile loop.

* Device Op GroupedGEMMTileLoop.

* Bunch of small changes in exmaple.

* CkProfiler

* Remove unused tparam.

* Fix include statement.

* Fix output stream overloads.

* Do not make descriptors and check validity untill we find group.

* Fix gemm desc initialization.

* Revert device op

* Fix compilation for DTYPES=FP16

* Validate tensor transfers paramters.

* Validate on host only NK dims if M is not known.

* Fix bug.

* A convenient debug func for selecting threads.

* Fix has main k block loop bug.

* Make sure that b2c has up to date tile offset.

* Output stream operator for Sequence type.

* Cmake file formatting.

2024-04-25 15:12:53 -05:00

..

CMakeLists.txt

Grouped GEMM Multiple D tile loop. (#1247)

2024-04-25 15:12:53 -05:00

test_groupnorm_bwd_data_fp32.cpp

layernorm and groupnorm backward data (#1083)

2023-12-19 04:23:11 +08:00

test_layernorm2d_bwd_data_fp32.cpp

layernorm and groupnorm backward data (#1083)

2023-12-19 04:23:11 +08:00