Unverified Commit 5023bd27 authored by Vincent QB's avatar Vincent QB Committed by GitHub
Browse files

Revert "Revert "Change Tensor.data<T> to Tensor.data_ptr<T> (#261)" (#265)" (#312)

This reverts commit 962c6b0f.
parent 8920802d
......@@ -37,7 +37,7 @@ int64_t write_audio(SoxDescriptor& fd, at::Tensor tensor) {
std::vector<sox_sample_t> buffer(tensor.numel());
AT_DISPATCH_ALL_TYPES(tensor.scalar_type(), "write_audio_buffer", [&] {
auto* data = tensor.data<scalar_t>();
auto* data = tensor.data_ptr<scalar_t>();
std::copy(data, data + tensor.numel(), buffer.begin());
});
......@@ -64,7 +64,7 @@ void read_audio(
output = output.contiguous();
AT_DISPATCH_ALL_TYPES(output.scalar_type(), "read_audio_buffer", [&] {
auto* data = output.data<scalar_t>();
auto* data = output.data_ptr<scalar_t>();
std::copy(buffer.begin(), buffer.begin() + samples_read, data);
});
}
......@@ -370,7 +370,7 @@ int build_flow_effects(const std::string& file_name,
const int64_t samples_read = sox_read(input, samples.data(), buffer_size);
assert(samples_read != nc * ns && samples_read != 0);
AT_DISPATCH_ALL_TYPES(otensor.scalar_type(), "effects_buffer", [&] {
auto* data = otensor.data<scalar_t>();
auto* data = otensor.data_ptr<scalar_t>();
std::copy(samples.begin(), samples.begin() + samples_read, data);
});
// free buffer and close mem_read
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment