Merge pull request rapidsai#14603 from rapidsai/branch-23.12

[HOTFIX] `v23.12.01`
NvTimLiu · Dec 8, 2023 · fba06e9 · fba06e9
2 parents c1d3073 + fd2f6a6
commit fba06e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cpp/src/io/orc/writer_impl.cu b/cpp/src/io/orc/writer_impl.cu
@@ -2137,7 +2137,8 @@ stripe_dictionaries build_dictionaries(orc_table_view& orc_table,
       }
     }
   }
-  stripe_dicts.host_to_device_async(stream);
+  // Synchronize to ensure the copy is complete before we clear `map_slots`
+  stripe_dicts.host_to_device_sync(stream);
 
   gpu::collect_map_entries(stripe_dicts, stream);
   gpu::get_dictionary_indices(stripe_dicts, orc_table.d_columns, stream);