From 2b0f7be7fd8739e099787e621e80c7e5fa85ccd8 Mon Sep 17 00:00:00 2001
From: Aayush Gupta <aaygupta@nvidia.com>
Date: Wed, 26 Jun 2024 22:13:25 +0000
Subject: [PATCH] Smaller FFT test sizes for faster CI/CD

---
 test/00_transform/FFT.cu | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/test/00_transform/FFT.cu b/test/00_transform/FFT.cu
index d77a3786..7056feb2 100644
--- a/test/00_transform/FFT.cu
+++ b/test/00_transform/FFT.cu
@@ -162,8 +162,8 @@ TYPED_TEST(FFTTestComplexNonHalfTypesAllExecs, FFT1Axis)
   using TestType = cuda::std::tuple_element_t<0, TypeParam>;
 
   const int d1 = 8;
-  const int d2 = 512;
-  const int d3 = 1024;
+  const int d2 = 32;
+  const int d3 = 64;
 
   // example-begin fft-2
   auto in = make_tensor<TestType>({d1, d2, d3});
@@ -280,9 +280,9 @@ TYPED_TEST(FFTTestComplexNonHalfTypesAllExecs, FFT2Axis)
   MATX_ENTER_HANDLER();
   using TestType = cuda::std::tuple_element_t<0, TypeParam>;
 ; 
-  const int d1 = 128;
-  const int d2 = 256;
-  const int d3 = 512;
+  const int d1 = 8;
+  const int d2 = 16;
+  const int d3 = 32;
 
   auto in = make_tensor<TestType>({d1, d2, d3});
   auto out1 = make_tensor<TestType>({d1, d2, d3});