Skip to content

Commit b132f0f

Browse files
gchananfacebook-github-bot
authored andcommitted
Remove 'Tensor' key from ATen codegen. (pytorch#17782)
Summary: We used to have different ATen Tensor types, but we don't anymore. This was just being maintained by a codegen'ed comment. Pull Request resolved: pytorch#17782 Reviewed By: ezyang Differential Revision: D14378004 Pulled By: gchanan fbshipit-source-id: 1bbf276393a391252d372cc385230c784bd78588
1 parent 5ffd7db commit b132f0f

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

aten/src/ATen/gen.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -263,8 +263,6 @@ def generate_storage_type_and_tensor(backend, density, scalar_type, declarations
263263
env['AccScalarName'] = accreal
264264
env['isFloatingType'] = is_floating_type
265265
env['isIntegralType'] = not is_floating_type
266-
if density == 'Dense':
267-
env['Tensor'] = "{}{}{}Tensor".format(density_tag, backend, scalar_name)
268266
env['Type'] = "{}{}{}Type".format(density_tag, backend, scalar_name)
269267
env['DenseTensor'] = "{}{}Tensor".format(backend, scalar_name)
270268
env['Backend'] = density_tag + backend

aten/src/ATen/templates/TypeDerived.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ size_t ${Type}::elementSizeInBytes() const {
5959

6060
/* example
6161
Tensor * ${Type}::add(Tensor & a, Tensor & b) {
62-
std::cout << "add ${Tensor}\n";
62+
std::cout << "add Tensor with backend ${Backend}\n";
6363
return &a;
6464
}
6565
*/

0 commit comments

Comments
 (0)