@@ -351,13 +351,13 @@ template struct delete_memory_op<std::complex<double>, base_device::DEVICE_GPU>;
351351template <typename FPTYPE>
352352struct resize_memory_op <FPTYPE, base_device::DEVICE_DSP>
353353{
354- void operator ()(const base_device::DEVICE_DSP * dev, FPTYPE*& arr, const size_t size, const char * record_in)
354+ void operator ()(const base_device::DEVICE_CPU * dev, FPTYPE*& arr, const size_t size, const char * record_in)
355355 {
356356 if (arr != nullptr )
357357 {
358358 free_ht (arr);
359359 }
360- arr = (FPTYPE*)malloc_ht (sizeof (FPTYPE) * size);
360+ arr = (FPTYPE*)malloc_ht (sizeof (FPTYPE) * size, GlobalV::MY_RANK );
361361 std::string record_string;
362362 if (record_in != nullptr )
363363 {
@@ -378,7 +378,7 @@ struct resize_memory_op<FPTYPE, base_device::DEVICE_DSP>
378378template <typename FPTYPE>
379379struct synchronize_memory_op <FPTYPE, base_device::DEVICE_DSP, base_device::DEVICE_CPU>
380380{
381- void operator ()(const base_device::DEVICE_DSP * dev_out,
381+ void operator ()(const base_device::DEVICE_CPU * dev_out,
382382 const base_device::DEVICE_CPU* dev_in,
383383 FPTYPE* arr_out,
384384 const FPTYPE* arr_in,
@@ -396,7 +396,7 @@ template <typename FPTYPE>
396396struct synchronize_memory_op <FPTYPE, base_device::DEVICE_CPU, base_device::DEVICE_DSP>
397397{
398398 void operator ()(const base_device::DEVICE_CPU* dev_out,
399- const base_device::DEVICE_DSP * dev_in,
399+ const base_device::DEVICE_CPU * dev_in,
400400 FPTYPE* arr_out,
401401 const FPTYPE* arr_in,
402402 const size_t size)
@@ -412,8 +412,8 @@ struct synchronize_memory_op<FPTYPE, base_device::DEVICE_CPU, base_device::DEVIC
412412template <typename FPTYPE>
413413struct synchronize_memory_op <FPTYPE, base_device::DEVICE_DSP, base_device::DEVICE_DSP>
414414{
415- void operator ()(const base_device::DEVICE_DSP * dev_out,
416- const base_device::DEVICE_DSP * dev_in,
415+ void operator ()(const base_device::DEVICE_CPU * dev_out,
416+ const base_device::DEVICE_CPU * dev_in,
417417 FPTYPE* arr_out,
418418 const FPTYPE* arr_in,
419419 const size_t size)
@@ -431,7 +431,7 @@ template <typename FPTYPE_out, typename FPTYPE_in>
431431struct cast_memory_op <FPTYPE_out, FPTYPE_in, base_device::DEVICE_CPU, base_device::DEVICE_DSP>
432432{
433433 void operator ()(const base_device::DEVICE_CPU* dev_out,
434- const base_device::DEVICE_DSP * dev_in,
434+ const base_device::DEVICE_CPU * dev_in,
435435 FPTYPE_out* arr_out,
436436 const FPTYPE_in* arr_in,
437437 const size_t size)
@@ -449,8 +449,8 @@ struct cast_memory_op<FPTYPE_out, FPTYPE_in, base_device::DEVICE_CPU, base_devic
449449template <typename FPTYPE_out, typename FPTYPE_in>
450450struct cast_memory_op <FPTYPE_out, FPTYPE_in, base_device::DEVICE_DSP, base_device::DEVICE_DSP>
451451{
452- void operator ()(const base_device::DEVICE_DSP * dev_out,
453- const base_device::DEVICE_DSP * dev_in,
452+ void operator ()(const base_device::DEVICE_CPU * dev_out,
453+ const base_device::DEVICE_CPU * dev_in,
454454 FPTYPE_out* arr_out,
455455 const FPTYPE_in* arr_in,
456456 const size_t size)
@@ -468,7 +468,7 @@ struct cast_memory_op<FPTYPE_out, FPTYPE_in, base_device::DEVICE_DSP, base_devic
468468template <typename FPTYPE_out, typename FPTYPE_in>
469469struct cast_memory_op <FPTYPE_out, FPTYPE_in, base_device::DEVICE_DSP, base_device::DEVICE_CPU>
470470{
471- void operator ()(const base_device::DEVICE_DSP * dev_out,
471+ void operator ()(const base_device::DEVICE_CPU * dev_out,
472472 const base_device::DEVICE_CPU* dev_in,
473473 FPTYPE_out* arr_out,
474474 const FPTYPE_in* arr_in,
@@ -487,7 +487,7 @@ struct cast_memory_op<FPTYPE_out, FPTYPE_in, base_device::DEVICE_DSP, base_devic
487487template <typename FPTYPE>
488488struct delete_memory_op <FPTYPE, base_device::DEVICE_DSP>
489489{
490- void operator ()(const base_device::DEVICE_DSP * dev, FPTYPE* arr)
490+ void operator ()(const base_device::DEVICE_CPU * dev, FPTYPE* arr)
491491 {
492492 free_ht (arr);
493493 }
0 commit comments