Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactored buffer copying to avoid wrong usage of buffers.

master
shugeo 2019-10-02 16:51:09 +03:00
parent 1c6173d218
commit f3e42173ef
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libnd4j/include/ops/declarable/generic/datatypes/bitcast.cpp
View File

@ -29,20 +29,15 @@ namespace nd4j {
CUSTOM_OP_IMPL(bitcast, 1, 1, false, 0, 1) { CUSTOM_OP_IMPL(bitcast, 1, 1, false, 0, 1) {
auto input = INPUT_VARIABLE(0); auto input = INPUT_VARIABLE(0);
auto output = OUTPUT_VARIABLE(0); auto output = OUTPUT_VARIABLE(0);
// when empty - nothing to do
if(input->isEmpty()){ if(input->isEmpty()){
REQUIRE_TRUE(output->isEmpty(), 0, "BITCAST: If input is empty, output array must also be empty."); REQUIRE_TRUE(output->isEmpty(), 0, "BITCAST: If input is empty, output array must also be empty.");
return Status::OK(); return Status::OK();
} }
// buffers for both input and output should be equals
DataBuffer buf(input->buffer(), input->specialBuffer(), input->lengthOf() * input->sizeOfT(), input->dataType());
*(output->dataBuffer()) = buf;
// if (!block.isInplace())
// output->assign(input);
input->syncToHost();
output->syncToHost();
memcpy(output->buffer(), input->buffer(), input->lengthOf() * input->sizeOfT());
output->syncToDevice();
output->tickWriteDevice();
//STORE_RESULT(output);
return Status::OK(); return Status::OK();
} }
DECLARE_SYN(BitCast, bitcast); DECLARE_SYN(BitCast, bitcast);