Skip to content

Commit 6ce9d65

Browse files
perf: fewer copies
1 parent d2dc658 commit 6ce9d65

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

dijkstra3d.pyx

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -772,7 +772,7 @@ def _execute_value_target_dijkstra(
772772
voxel_graph_ptr
773773
)
774774
elif dtype in (np.int64, np.uint64):
775-
arr_memview64 = data.astype(np.uint64)
775+
arr_memview64 = data.view(np.uint64)
776776
if sixtyfourbit:
777777
output64 = value_target_dijkstra3d[uint64_t, uint64_t](
778778
&arr_memview64[0,0,0],
@@ -788,7 +788,7 @@ def _execute_value_target_dijkstra(
788788
voxel_graph_ptr
789789
)
790790
elif dtype in (np.int32, np.uint32):
791-
arr_memview32 = data.astype(np.uint32)
791+
arr_memview32 = data.view(np.uint32)
792792
if sixtyfourbit:
793793
output64 = value_target_dijkstra3d[uint32_t, uint64_t](
794794
&arr_memview32[0,0,0],
@@ -804,7 +804,7 @@ def _execute_value_target_dijkstra(
804804
voxel_graph_ptr
805805
)
806806
elif dtype in (np.int16, np.uint16):
807-
arr_memview16 = data.astype(np.uint16)
807+
arr_memview16 = data.view(np.uint16)
808808
if sixtyfourbit:
809809
output64 = value_target_dijkstra3d[uint16_t, uint64_t](
810810
&arr_memview16[0,0,0],
@@ -820,7 +820,7 @@ def _execute_value_target_dijkstra(
820820
voxel_graph_ptr
821821
)
822822
elif dtype in (np.int8, np.uint8, bool):
823-
arr_memview8 = data.astype(np.uint8)
823+
arr_memview8 = data.view(np.uint8)
824824
if sixtyfourbit:
825825
output64 = value_target_dijkstra3d[uint8_t, uint64_t](
826826
&arr_memview8[0,0,0],
@@ -1607,31 +1607,31 @@ def _execute_distance_field(data, sources, connectivity, voxel_graph):
16071607
voxel_graph_ptr, max_loc
16081608
)
16091609
elif dtype in (np.int64, np.uint64):
1610-
arr_memview64 = data.astype(np.uint64)
1610+
arr_memview64 = data.view(np.uint64)
16111611
dist = distance_field3d[uint64_t](
16121612
&arr_memview64[0,0,0],
16131613
sx, sy, sz,
16141614
src, connectivity,
16151615
voxel_graph_ptr, max_loc
16161616
)
16171617
elif dtype in (np.uint32, np.int32):
1618-
arr_memview32 = data.astype(np.uint32)
1618+
arr_memview32 = data.view(np.uint32)
16191619
dist = distance_field3d[uint32_t](
16201620
&arr_memview32[0,0,0],
16211621
sx, sy, sz,
16221622
src, connectivity,
16231623
voxel_graph_ptr, max_loc
16241624
)
16251625
elif dtype in (np.int16, np.uint16):
1626-
arr_memview16 = data.astype(np.uint16)
1626+
arr_memview16 = data.view(np.uint16)
16271627
dist = distance_field3d[uint16_t](
16281628
&arr_memview16[0,0,0],
16291629
sx, sy, sz,
16301630
src, connectivity,
16311631
voxel_graph_ptr, max_loc
16321632
)
16331633
elif dtype in (np.int8, np.uint8, bool):
1634-
arr_memview8 = data.astype(np.uint8)
1634+
arr_memview8 = data.view(np.uint8)
16351635
dist = distance_field3d[uint8_t](
16361636
&arr_memview8[0,0,0],
16371637
sx, sy, sz,
@@ -1720,7 +1720,7 @@ def _execute_parental_field(data, source, connectivity, voxel_graph):
17201720
voxel_graph_ptr
17211721
)
17221722
elif dtype in (np.int64, np.uint64):
1723-
arr_memview64 = data.astype(np.uint64)
1723+
arr_memview64 = data.view(np.uint64)
17241724
if sixtyfourbit:
17251725
parental_field3d[uint64_t,uint64_t](
17261726
&arr_memview64[0,0,0],
@@ -1738,7 +1738,7 @@ def _execute_parental_field(data, source, connectivity, voxel_graph):
17381738
voxel_graph_ptr
17391739
)
17401740
elif dtype in (np.uint32, np.int32):
1741-
arr_memview32 = data.astype(np.uint32)
1741+
arr_memview32 = data.view(np.uint32)
17421742
if sixtyfourbit:
17431743
parental_field3d[uint32_t,uint64_t](
17441744
&arr_memview32[0,0,0],
@@ -1756,7 +1756,7 @@ def _execute_parental_field(data, source, connectivity, voxel_graph):
17561756
voxel_graph_ptr
17571757
)
17581758
elif dtype in (np.int16, np.uint16):
1759-
arr_memview16 = data.astype(np.uint16)
1759+
arr_memview16 = data.view(np.uint16)
17601760
if sixtyfourbit:
17611761
parental_field3d[uint16_t,uint64_t](
17621762
&arr_memview16[0,0,0],
@@ -1774,7 +1774,7 @@ def _execute_parental_field(data, source, connectivity, voxel_graph):
17741774
voxel_graph_ptr
17751775
)
17761776
elif dtype in (np.int8, np.uint8, bool):
1777-
arr_memview8 = data.astype(np.uint8)
1777+
arr_memview8 = data.view(np.uint8)
17781778
if sixtyfourbit:
17791779
parental_field3d[uint8_t,uint64_t](
17801780
&arr_memview8[0,0,0],

0 commit comments

Comments
 (0)