Skip to content

Commit 9d45b84

Browse files
authored
feat: set enzyme comm flag (#1861)
1 parent 81926ba commit 9d45b84

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

deps/ReactantExtra/API.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1208,6 +1208,7 @@ GenerateCompileOptions(int64_t device_id, const int64_t *mesh_ids,
12081208
auto debug_options = options.executable_build_options.mutable_debug_options();
12091209

12101210
debug_options->set_xla_gpu_cuda_data_dir(xla_gpu_cuda_data_dir);
1211+
debug_options->set_xla_enable_enzyme_comms_opt(true);
12111212

12121213
if (kernel_cache_enabled) {
12131214
debug_options->set_xla_gpu_kernel_cache_file(kernel_cache_path);

0 commit comments

Comments
 (0)