File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments