@@ -661,7 +661,7 @@ void FlatVariables::assign_cb(drjit::TraversableBase *traversable) {
661
661
void FlatVariables::traverse (nb::handle h, TraverseContext &ctx) {
662
662
recursion_guard guard (this );
663
663
664
- scoped_set_flag traverse_scope (JitFlag::FreezingTraverseScope , true );
664
+ scoped_set_flag traverse_scope (JitFlag::EnableObjectTraversal , true );
665
665
666
666
ProfilerPhase profiler (" traverse" );
667
667
nb::handle tp = h.type ();
@@ -894,7 +894,7 @@ nb::object FlatVariables::construct() {
894
894
*/
895
895
void FlatVariables::assign (nb::handle dst) {
896
896
recursion_guard guard (this );
897
- scoped_set_flag traverse_scope (JitFlag::FreezingTraverseScope , true );
897
+ scoped_set_flag traverse_scope (JitFlag::EnableObjectTraversal , true );
898
898
899
899
nb::handle tp = dst.type ();
900
900
Layout &layout = this ->layout [layout_index++];
@@ -1020,7 +1020,7 @@ void FlatVariables::assign(nb::handle dst) {
1020
1020
* additional data to vcalls is tracked correctly.
1021
1021
*/
1022
1022
void FlatVariables::traverse_with_registry (nb::handle h, TraverseContext &ctx) {
1023
- scoped_set_flag traverse_scope (JitFlag::FreezingTraverseScope , true );
1023
+ scoped_set_flag traverse_scope (JitFlag::EnableObjectTraversal , true );
1024
1024
1025
1025
// Traverse the handle
1026
1026
traverse (h, ctx);
@@ -1078,7 +1078,7 @@ void FlatVariables::traverse_with_registry(nb::handle h, TraverseContext &ctx) {
1078
1078
* Corresponds to `traverse_with_registry`.
1079
1079
*/
1080
1080
void FlatVariables::assign_with_registry (nb::handle dst) {
1081
- scoped_set_flag traverse_scope (JitFlag::FreezingTraverseScope , true );
1081
+ scoped_set_flag traverse_scope (JitFlag::EnableObjectTraversal , true );
1082
1082
1083
1083
// Assign the handle
1084
1084
assign (dst);
0 commit comments