Skip to content

Commit 3408562

Browse files
committed
Make concatenate tuples python2-friendly
1 parent 11f0f65 commit 3408562

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

torch2c/emitters.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -110,13 +110,13 @@ def tensor_meta_tpl(size_name, stride_name, size, stride=None):
110110
dim = len(size)
111111

112112
if dim == 1:
113-
meta_size = 'THLongStorage *%s = THLongStorage_newWithSize1(%d);' % (size_name, *size)
113+
meta_size = 'THLongStorage *%s = THLongStorage_newWithSize1(%d);' % (size_name,) + size
114114
elif dim == 2:
115-
meta_size = 'THLongStorage *%s = THLongStorage_newWithSize2(%d,%d);' % (size_name, *size)
115+
meta_size = 'THLongStorage *%s = THLongStorage_newWithSize2(%d,%d);' % (size_name,) + size
116116
elif dim == 3:
117-
meta_size = 'THLongStorage *%s = THLongStorage_newWithSize3(%d,%d,%d);' % (size_name, *size)
117+
meta_size = 'THLongStorage *%s = THLongStorage_newWithSize3(%d,%d,%d);' % (size_name,) + size
118118
elif dim == 4:
119-
meta_size = 'THLongStorage *%s = THLongStorage_newWithSize4(%d,%d,%d,%d);' % (size_name, *size)
119+
meta_size = 'THLongStorage *%s = THLongStorage_newWithSize4(%d,%d,%d,%d);' % (size_name,) + size
120120
else:
121121
meta_size_lines = []
122122
meta_size_lines.append(
@@ -130,13 +130,13 @@ def tensor_meta_tpl(size_name, stride_name, size, stride=None):
130130

131131
if stride:
132132
if dim == 1:
133-
meta_stride = 'THLongStorage *%s = THLongStorage_newWithSize1(%d);' % (stride_name, *stride)
133+
meta_stride = 'THLongStorage *%s = THLongStorage_newWithSize1(%d);' % (stride_name,) + stride
134134
elif dim == 2:
135-
meta_stride = 'THLongStorage *%s = THLongStorage_newWithSize2(%d,%d);' % (stride_name, *stride)
135+
meta_stride = 'THLongStorage *%s = THLongStorage_newWithSize2(%d,%d);' % (stride_name,) + stride
136136
elif dim == 3:
137-
meta_stride = 'THLongStorage *%s = THLongStorage_newWithSize3(%d,%d,%d);' % (stride_name, *stride)
137+
meta_stride = 'THLongStorage *%s = THLongStorage_newWithSize3(%d,%d,%d);' % (stride_name,) + stride
138138
elif dim == 4:
139-
meta_stride = 'THLongStorage *%s = THLongStorage_newWithSize4(%d,%d,%d,%d);' % (stride_name, *stride)
139+
meta_stride = 'THLongStorage *%s = THLongStorage_newWithSize4(%d,%d,%d,%d);' % (stride_name,) + stride
140140
else:
141141
meta_stride_lines = []
142142
meta_stride_lines.append(

0 commit comments

Comments
 (0)