CUTLASS 2.4 (Implicit GEMM convolution) (#147)

CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-19 21:25:25 -08:00
parent c2b80ad4e4
commit 6615010cd0
224 changed files with 43939 additions and 1061 deletions
@@ -36,6 +36,11 @@ namespace cutlass {
 namespace library {

 //////////////////////////////////////////////////////////////////////////////////////////////////////////
+
+void initialize_reference_operations(Manifest &manifest);
+
+//////////////////////////////////////////////////////////////////////////////////////////////////////////
+
 /// Top-level initialization
 Status Manifest::initialize() {

@@ -46,6 +51,12 @@ Status Manifest::initialize() {
  // initialize procedurally generated cutlass op in manifest object
  initialize_all(*this);

+  // initialize manually instanced conv3d reference op in manifest object
+  initialize_reference_operations(*this);
+
+  // initialize manually instanced reduction reference op in manifest object
+  initialize_all_reduction_op(*this);
+
  return Status::kSuccess;
 }