@@ -110,13 +110,13 @@ def tensor_meta_tpl(size_name, stride_name, size, stride=None):
110
110
dim = len (size )
111
111
112
112
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
114
114
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
116
116
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
118
118
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
120
120
else :
121
121
meta_size_lines = []
122
122
meta_size_lines .append (
@@ -130,13 +130,13 @@ def tensor_meta_tpl(size_name, stride_name, size, stride=None):
130
130
131
131
if stride :
132
132
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
134
134
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
136
136
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
138
138
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
140
140
else :
141
141
meta_stride_lines = []
142
142
meta_stride_lines .append (
0 commit comments