frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-22 14:15:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Deprecate tensor.data<T>(), and codemod tensor.data<T>() to tensor.data_ptr<T>() (#24886)

Browse Source 
Summary:
This PR adds deprecation message for `tensor.data<T>()` (91d94e7d41), and changes all call sites of `tensor.data<T>()` to `tensor.data_ptr<T>()`  in PyTorch core.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/24886

Differential Revision: D16924576

Pulled By: yf225

fbshipit-source-id: 0943d6be73245c7c549c78597b74c3b07fa24440
This commit is contained in:
Will Feng
2019-08-21 20:09:37 -07:00
committed by Facebook Github Bot
parent aa66146974
commit 420b37f3c6
135 changed files with 989 additions and 987 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test/cpp/api/module.cpp
View File

@ -655,8 +655,8 @@ TEST_F(ModuleTest, ParametersReturnsExpectedTensorsForFlatModel) {
TestModule module(1);
std::vector<torch::Tensor> parameters = module.parameters();
ASSERT_EQ(parameters.size(), 2);
ASSERT_EQ(parameters[0].data<float>(), module.p1.data<float>());
ASSERT_EQ(parameters[1].data<float>(), module.p2.data<float>());
ASSERT_EQ(parameters[0].data_ptr<float>(), module.p1.data_ptr<float>());
ASSERT_EQ(parameters[1].data_ptr<float>(), module.p2.data_ptr<float>());
}
TEST_F(ModuleTest, NamedParametersReturnsExpectedTensorsForFlatModel) {
@ -665,17 +665,17 @@ TEST_F(ModuleTest, NamedParametersReturnsExpectedTensorsForFlatModel) {
module.named_parameters();
ASSERT_EQ(parameters.size(), 2);
ASSERT_EQ(parameters[0].key(), "p1");
ASSERT_EQ(parameters[0]->data<float>(), module.p1.data<float>());
ASSERT_EQ(parameters[0]->data_ptr<float>(), module.p1.data_ptr<float>());
ASSERT_EQ(parameters[1].key(), "p2");
ASSERT_EQ(parameters[1]->data<float>(), module.p2.data<float>());
ASSERT_EQ(parameters[1]->data_ptr<float>(), module.p2.data_ptr<float>());
}
TEST_F(ModuleTest, BuffersReturnsExpectedTensorsForFlatModel) {
TestModule module(1);
std::vector<torch::Tensor> buffers = module.buffers();
ASSERT_EQ(buffers.size(), 2);
ASSERT_EQ(buffers[0].data<float>(), module.b1.data<float>());
ASSERT_EQ(buffers[1].data<float>(), module.b2.data<float>());
ASSERT_EQ(buffers[0].data_ptr<float>(), module.b1.data_ptr<float>());
ASSERT_EQ(buffers[1].data_ptr<float>(), module.b2.data_ptr<float>());
}
TEST_F(ModuleTest, NamedBuffersReturnsExpectedTensorsForFlatModel) {
@ -684,9 +684,9 @@ TEST_F(ModuleTest, NamedBuffersReturnsExpectedTensorsForFlatModel) {
module.named_buffers();
ASSERT_EQ(buffers.size(), 2);
ASSERT_EQ(buffers[0].key(), "b1");
ASSERT_EQ(buffers[0]->data<float>(), module.b1.data<float>());
ASSERT_EQ(buffers[0]->data_ptr<float>(), module.b1.data_ptr<float>());
ASSERT_EQ(buffers[1].key(), "b2");
ASSERT_EQ(buffers[1]->data<float>(), module.b2.data<float>());
ASSERT_EQ(buffers[1]->data_ptr<float>(), module.b2.data_ptr<float>());
}
struct TestContainer : torch::nn::Module {