[Bug Fix] GLM-5.1: drop constexpr on page_indice_batch_offset, skip offloader post_init on draft worker, support N=32 in copy_to_gpu_no_ce (#23550)

2026-05-09 15:43:45 +08:00
parent 9d12f9e6fa
commit d49fc092cb
4 changed files with 151 additions and 2 deletions
@@ -52,6 +52,8 @@ void copy_to_gpu_no_ce(const at::Tensor& input, at::Tensor& output) {
    copy_to_gpu_no_ce_impl<72>(input, output);
  } else if (N == 64) {
    copy_to_gpu_no_ce_impl<64>(input, output);
+  } else if (N == 32) {
+    copy_to_gpu_no_ce_impl<32>(input, output);
  } else {
    TORCH_CHECK(false, "unexpected N");
  }