Fix OSS ubuntu test failure (pytorch#653)

Summary: Pull Request resolved: pytorch#653 OSS ubuntu tests were getting killed due to longer test time. Reduce the number of shapes to just the interesting ones. Reviewed By: jspark1105 Differential Revision: D29797295 fbshipit-source-id: d914e75bc74cf6419bf6a3463df7abc153e3c74d
pls331 · Jul 21, 2021 · d2440b4 · d2440b4
1 parent bef192a
commit d2440b4
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 15 deletions.
diff --git a/test/EmbeddingSpMDMTest.cc b/test/EmbeddingSpMDMTest.cc
@@ -25,11 +25,11 @@ static vector<vector<int>> GetInputs_() {
       // batch size, number of rows of table, emb dim , avg length
       {1, 8, 8, 4},
       {10, 4000, 32, 100},
-      {100, 4000, 32, 100},
+      // {100, 4000, 32, 100},
       {10, 4000, 128, 100},
       {4, 400, 256, 10},
       {10, 4000, 48, 100},
-      {10, 4000, 40, 100},
+      // {10, 4000, 40, 100},
       {10, 4000, 56, 100},
       {10, 4000, 1, 100},
       // These were  from C2 tests

diff --git a/test/I8DepthwiseTest.cc b/test/I8DepthwiseTest.cc
@@ -43,7 +43,7 @@ static vector<vector<int>> shapes = {
 //  {   1,  544,  38,  40, 1, 3 },
   {   1,  544,  47,  38, 1, 3 },
   {   1, 1088,   7,   7, 1, 3 },
-  {  51, 1088,   7,   7, 1, 3 },
+  {  2, 1088,   7,   7, 1, 3 },
   {   2, 1088,   7,   7, 1, 5 },
 //  { 100, 1088,   7,   7, 1, 3 },
 
@@ -64,7 +64,7 @@ static vector<vector<int>> shapes = {
 //  {   1,  272,  75,  76, 2, 3 },
   {   1,  272,  94,  75, 2, 3 },
   {   1,  544,  14,  14, 2, 3 },
-  {  51,  544,  14,  14, 2, 3 },
+  // {  51,  544,  14,  14, 2, 3 },
 //  { 100,  544,  14,  14, 2, 3 },
 
   {   1,    8,   4,   4, 1, 3 },

diff --git a/test/PackedRequantizeTest.cc b/test/PackedRequantizeTest.cc
@@ -75,26 +75,26 @@ static vector<vector<int>> GetShapes_() {
   vector<vector<int>> shapes = {
     // {M,    N,    K}
     {1, 128, 512},
-    {1, 1024, 256},
-    {1, 2048, 512},
+    // {1, 1024, 256},
+    // {1, 2048, 512},
     {1, 2048, 513},
-    {1, 2048, 514},
+    // {1, 2048, 514},
 
     {6, 512, 512},
-    {6, 2048, 512},
-    {6, 256, 1024},
-    {6, 1024, 256},
+    // {6, 2048, 512},
+    // {6, 256, 1024},
+    // {6, 1024, 256},
     {6, 2048, 256},
     {6, 2048, 257},
-    {6, 2048, 258},
+    // {6, 2048, 258},
 
-    {102, 1024, 512},
+    // {102, 1024, 512},
     {102, 2323, 256},
-    {102, 512, 256},
+    // {102, 512, 256},
     {102, 512, 257},
-    {102, 512, 258},
+    // {102, 512, 258},
 
-    {1024, 512, 258},
+    // {1024, 512, 258},
 
     {120, 4, 288},
   };