Skip to content

Commit 071fce7

Browse files
committed
Fix template bug
1 parent 02ca03c commit 071fce7

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

source/module_base/module_device/memory_op.cpp

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -351,13 +351,13 @@ template struct delete_memory_op<std::complex<double>, base_device::DEVICE_GPU>;
351351
template <typename FPTYPE>
352352
struct 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>
378378
template <typename FPTYPE>
379379
struct 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>
396396
struct 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
412412
template <typename FPTYPE>
413413
struct 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>
431431
struct 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
449449
template <typename FPTYPE_out, typename FPTYPE_in>
450450
struct 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
468468
template <typename FPTYPE_out, typename FPTYPE_in>
469469
struct 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
487487
template <typename FPTYPE>
488488
struct 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

Comments
 (0)