@@ -133,40 +133,6 @@ bool VarLayout::operator==(const VarLayout &rhs) const {
133
133
return true ;
134
134
}
135
135
136
- static void log_layouts (const std::vector<Layout> &layouts, std::ostream &os,
137
- uint32_t &index, std::string &padding) {
138
- const Layout &layout = layouts[index ++];
139
-
140
- auto tp_name = layout.type ? nb::type_name (layout.type ).c_str () : " None" ;
141
- os << padding << " type = " << tp_name << std::endl;
142
- os << padding << " num: " << layout.num << std::endl;
143
- os << padding << " flags: " << std::bitset<8 >(layout.flags ) << std::endl;
144
- os << padding << " index: " << layout.index << std::endl;
145
- os << padding << " py_object: " << nb::str (layout.py_object ).c_str ()
146
- << std::endl;
147
-
148
- if (layout.fields .size () == 0 )
149
- for (uint32_t i = 0 ; i < layout.num ; i++) {
150
- os << padding << " Layout[" << std::endl;
151
- padding.append (" " );
152
-
153
- log_layouts (layouts, os, index , padding);
154
-
155
- padding.resize (padding.length () - 4 );
156
- os << padding << " ]" << std::endl;
157
- }
158
- else
159
- for (const auto &field : layout.fields ) {
160
- os << padding << nb::str (field).c_str () << " : Layout[" << std::endl;
161
- padding.append (" " );
162
-
163
- log_layouts (layouts, os, index , padding);
164
-
165
- padding.resize (padding.length () - 4 );
166
- os << padding << " ]" << std::endl;
167
- }
168
- }
169
-
170
136
/* *
171
137
* \brief Add a variant domain pair to be traversed using the registry.
172
138
*
@@ -402,7 +368,7 @@ void FlatVariables::traverse_ad_index(uint64_t index, TraverseContext &ctx,
402
368
// NOTE: instead of emplacing a Layout representing the ad variable always,
403
369
// we only do so if the gradients have been enabled. We use this format,
404
370
// since most variables will not be ad enabled. The layout therefore has to
405
- // be peeked in ``construct_ad_index`` before descending if an ad or jit
371
+ // be peeked in ``construct_ad_index`` before deciding if an ad or jit
406
372
// index should be constructed/assigned.
407
373
int grad_enabled = ad_grad_enabled (index );
408
374
if (grad_enabled) {
@@ -1145,38 +1111,6 @@ void FlatVariables::assign_with_registry(nb::handle dst) {
1145
1111
}
1146
1112
}
1147
1113
1148
- std::ostream &operator <<(std::ostream &os, const FlatVariables &r) {
1149
- std::string offset = " " ;
1150
-
1151
- os << " FlatVariables[" << std::endl;
1152
-
1153
- std::string padding (" " );
1154
- uint32_t index = 0 ;
1155
-
1156
- os << padding << " variables = [" ;
1157
- for (uint64_t index : r.variables ) {
1158
- os << " r%u, " ;
1159
- }
1160
- os << " ]" << std::endl;
1161
-
1162
- os << padding << " sizes = [" ;
1163
- for (uint64_t index : r.variables ) {
1164
- os << " %u, " ;
1165
- }
1166
- os << " ]" << std::endl;
1167
-
1168
- os << padding << " Layout[" << std::endl;
1169
-
1170
- padding.append (" " );
1171
- log_layouts (r.layout , os, index , padding);
1172
- padding.resize (padding.length () - 4 );
1173
-
1174
- os << padding << " ]" << std::endl;
1175
-
1176
- os << " ]" << std::endl;
1177
- return os;
1178
- }
1179
-
1180
1114
inline void hash_combine (size_t &seed, size_t value) {
1181
1115
// / From CityHash (https://github.com/google/cityhash)
1182
1116
const size_t mult = 0x9ddfea08eb382d69ull ;
0 commit comments