Skip to content

Commit e741164

Browse files
committed
patch CUDA/ipp/dask braggvects assign bug
Former-commit-id: 39a3956
1 parent e400add commit e741164

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

py4DSTEM/braggvectors/diskdetection.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -555,6 +555,8 @@ def _find_Bragg_disks_CUDA_unbatched(
555555
# Populate a BraggVectors instance and return
556556
braggvectors = BraggVectors(datacube.Rshape, datacube.Qshape)
557557
braggvectors._v_uncal = peaks
558+
braggvectors._set_raw_vector_getter()
559+
braggvectors._set_cal_vector_getter()
558560
return braggvectors
559561

560562

@@ -600,6 +602,8 @@ def _find_Bragg_disks_CUDA_batched(
600602
# Populate a BraggVectors instance and return
601603
braggvectors = BraggVectors(datacube.Rshape, datacube.Qshape)
602604
braggvectors._v_uncal = peaks
605+
braggvectors._set_raw_vector_getter()
606+
braggvectors._set_cal_vector_getter()
603607
return braggvectors
604608

605609

@@ -650,6 +654,8 @@ def _find_Bragg_disks_ipp(
650654
# Populate a BraggVectors instance and return
651655
braggvectors = BraggVectors(datacube.Rshape, datacube.Qshape)
652656
braggvectors._v_uncal = peaks
657+
braggvectors._set_raw_vector_getter()
658+
braggvectors._set_cal_vector_getter()
653659
return braggvectors
654660

655661

@@ -700,6 +706,8 @@ def _find_Bragg_disks_dask(
700706
# Populate a BraggVectors instance and return
701707
braggvectors = BraggVectors(datacube.Rshape, datacube.Qshape)
702708
braggvectors._v_uncal = peaks
709+
braggvectors._set_raw_vector_getter()
710+
braggvectors._set_cal_vector_getter()
703711
return braggvectors
704712

705713

0 commit comments

Comments
 (0)