tx258/lammps-gran-kokkos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix tabs->spaces in last commit

Browse Source
This commit is contained in:
Forrest Glines
2025-03-25 15:13:11 -07:00
parent 3c055fe93b
commit a98a743f2e
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/KOKKOS/mliap_unified_couple_kokkos.pyx
View File

@ -206,10 +206,10 @@ cdef class MLIAPDataPy:
copy_from_ptr = copy_from.data_ptr()
copy_to_ptr = copy_to.data_ptr()
if copy_from_dtype == torch.float32:
self.data.forward_exchange( <float*>copy_from_ptr, <float*>copy_to_ptr, vec_len)
elif copy_from_dtype == torch.float64:
self.data.forward_exchange( <double*>copy_from_ptr, <double*>copy_to_ptr, vec_len)
if copy_from_dtype == torch.float32:
self.data.forward_exchange( <float*>copy_from_ptr, <float*>copy_to_ptr, vec_len)
elif copy_from_dtype == torch.float64:
self.data.forward_exchange( <double*>copy_from_ptr, <double*>copy_to_ptr, vec_len)
else:
raise TypeError(f"Unsupported comms type: ({copy_from_dtype})")
except:
@ -237,10 +237,10 @@ cdef class MLIAPDataPy:
copy_from_ptr = copy_from.data_ptr()
copy_to_ptr = copy_to.data_ptr()
if copy_from_dtype == torch.float32:
self.data.reverse_exchange( <float*>copy_from_ptr, <float*>copy_to_ptr, vec_len)
elif copy_from_dtype == torch.float64:
self.data.reverse_exchange( <double*>copy_from_ptr, <double*>copy_to_ptr, vec_len)
if copy_from_dtype == torch.float32:
self.data.reverse_exchange( <float*>copy_from_ptr, <float*>copy_to_ptr, vec_len)
elif copy_from_dtype == torch.float64:
self.data.reverse_exchange( <double*>copy_from_ptr, <double*>copy_to_ptr, vec_len)
else:
raise TypeError(f"Unsupported comms type: ({copy_from_dtype})")
except: