split groupConv kernel from instruction specific code (pytorch#412)

Summary: Pull Request resolved: pytorch#412 split groupConv file into architecture-specifc and architecture-agnostic file Reviewed By: dskhudia Differential Revision: D23202000 fbshipit-source-id: 235fce2b265b90c9084272366cd84dfe2a43ce9a
zyan0 · Aug 19, 2020 · 685149b · 685149b
1 parent a3733e0
commit 685149b
Show file tree

Hide file tree

Showing 5 changed files with 1,613 additions and 1,579 deletions.
diff --git a/defs.bzl b/defs.bzl
@@ -29,6 +29,7 @@ def get_fbgemm_generic_srcs(with_base = False):
         "src/GenerateKernelU8S8S32ACC16Avx512VNNI.cc",
         "src/GenerateKernelU8S8S32ACC32.cc",
         "src/GenerateKernelU8S8S32ACC32Avx512VNNI.cc",
+        "src/GroupwiseConv.cc",
         "src/GroupwiseConvAcc32Avx2.cc",
         "src/PackAMatrix.cc",
         "src/PackAWithIm2Col.cc",