"vscode:/vscode.git/clone" did not exist on "96a2b54689e9458f6e03a21c2b5c0f573f31c573"
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) { ...@@ -37,7 +37,7 @@ int64_t write_audio(SoxDescriptor& fd, at::Tensor tensor) {
std::vector<sox_sample_t> buffer(tensor.numel()); std::vector<sox_sample_t> buffer(tensor.numel());
AT_DISPATCH_ALL_TYPES(tensor.scalar_type(), "write_audio_buffer", [&] { 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()); std::copy(data, data + tensor.numel(), buffer.begin());
}); });
...@@ -64,7 +64,7 @@ void read_audio( ...@@ -64,7 +64,7 @@ void read_audio(
output = output.contiguous(); output = output.contiguous();
AT_DISPATCH_ALL_TYPES(output.scalar_type(), "read_audio_buffer", [&] { 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); std::copy(buffer.begin(), buffer.begin() + samples_read, data);
}); });
} }
...@@ -370,7 +370,7 @@ int build_flow_effects(const std::string& file_name, ...@@ -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); const int64_t samples_read = sox_read(input, samples.data(), buffer_size);
assert(samples_read != nc * ns && samples_read != 0); assert(samples_read != nc * ns && samples_read != 0);
AT_DISPATCH_ALL_TYPES(otensor.scalar_type(), "effects_buffer", [&] { 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); std::copy(samples.begin(), samples.begin() + samples_read, data);
}); });
// free buffer and close mem_read // 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