From 5e5da19261e7d2dcaca0a341864a483d2d497234 Mon Sep 17 00:00:00 2001 From: 36000 Date: Sun, 21 Jan 2024 16:42:15 -0800 Subject: [PATCH] BFs --- AFQ/tractography/gputractography.py | 2 +- gpu_docker/cuda_track_template.def | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/AFQ/tractography/gputractography.py b/AFQ/tractography/gputractography.py index 4155344dc..c1461cf22 100644 --- 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 index 2d1170735..024f9436a 100644 --- 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 \