Skip to content

Commit 02ca03c

Browse files
committed
I finally find this
1 parent f188af4 commit 02ca03c

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

source/module_psi/psi.h

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,10 +143,17 @@ class Psi
143143

144144
bool allocate_inside = true; ///<whether allocate psi inside Psi class
145145

146+
#ifdef __DSP
147+
using set_memory_op = base_device::memory::set_memory_op<T, base_device::DEVICE_DSP>;
148+
using delete_memory_op = base_device::memory::delete_memory_op<T, base_device::DEVICE_DSP>;
149+
using resize_memory_op = base_device::memory::resize_memory_op<T, base_device::DEVICE_DSP>;
150+
using synchronize_memory_op = base_device::memory::synchronize_memory_op<T, base_device::DEVICE_DSP, base_device::DEVICE_DSP>;
151+
#else
146152
using set_memory_op = base_device::memory::set_memory_op<T, Device>;
147153
using delete_memory_op = base_device::memory::delete_memory_op<T, Device>;
148154
using resize_memory_op = base_device::memory::resize_memory_op<T, Device>;
149155
using synchronize_memory_op = base_device::memory::synchronize_memory_op<T, Device, Device>;
156+
#endif
150157
};
151158

152159
} // end of namespace psi

0 commit comments

Comments
 (0)