Fixed bug that occurs when non-c-contiguous arrays are passed to the …

…FastConv1d feature generator.
jlparkI · Oct 26, 2024 · 5d3c4cb · 5d3c4cb
1 parent 71d64a0
commit 5d3c4cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xGPR/kernels/convolution_kernels/conv_feature_extractor.py b/xGPR/kernels/convolution_kernels/conv_feature_extractor.py
@@ -111,12 +111,12 @@ def transform_x(self, input_x, sequence_length):
 
         if self.device == "cpu":
             output_x = np.zeros((input_x.shape[0], self.num_rffs), np.float64)
-            x_in = input_x.astype(np.float32, copy=False)
+            x_in = np.ascontiguousarray(input_x.astype(np.float32, copy=False))
             cpuConv1dMaxpool(x_in, output_x, self.radem_diag, self.chi_arr,
                     sequence_length, self.conv_width, self.num_threads, self.simplex_rffs)
         else:
             output_x = cp.zeros((input_x.shape[0], self.num_rffs), cp.float64)
-            x_in = cp.asarray(input_x).astype(cp.float32, copy=False)
+            x_in = cp.ascontiguousarray(cp.asarray(input_x).astype(cp.float32, copy=False))
             cudaConv1dMaxpool(x_in, output_x, self.radem_diag, self.chi_arr,
                     sequence_length, self.conv_width, self.simplex_rffs)