BFs

yeatmanlab · Jan 22, 2024 · 5e5da19 · 5e5da19
1 parent b612310
commit 5e5da19
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/AFQ/tractography/gputractography.py b/AFQ/tractography/gputractography.py
@@ -134,7 +134,7 @@ def gpu_track(data, gtab, seed_img, stop_img, odf_model,
             gtargs[var_name] = np.ascontiguousarray(
                 var, dtype=dtype)
         else:
-            gtargs[var_name] = var
+            gtargs[var_name] = np.asarray(var, dtype=dtype)
 
     gpu_tracker = cuslines.GPUTracker(
         model_type,

diff --git a/gpu_docker/cuda_track_template.def b/gpu_docker/cuda_track_template.def
@@ -23,7 +23,7 @@ From: nvidia/cuda:12.0.1-devel-ubuntu20.04
         ca-certificates gnupg software-properties-common gcc git wget numactl
 
     # Clone GPUStreamlines
-    git clone --recursive -b csaodf_debug_2 https://github.com/dipy/GPUStreamlines /opt/GPUStreamlines
+    git clone --recursive -b csaodf https://github.com/dipy/GPUStreamlines /opt/GPUStreamlines
 
     # Kitware repository for CMake
     wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null \