Skip to content

Commit d2596c7

Browse files
committed
Change mixed_D default value to false
1 parent 878d00f commit d2596c7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

fbgemm_gpu/codegen/training/backward/embedding_backward_dense_host_cpu.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ Tensor split_embedding_codegen_lookup_dense_function(
172172
c10::SymInt /* max_B = -1 */,
173173
c10::SymInt /* max_B_feature_rank = -1 */,
174174
c10::SymInt /* vbe_output_size = -1 */,
175-
bool /* mixed_D = true */) {
175+
bool /* mixed_D = false */) {
176176
return SplitLookupFunction_Dense_Op::apply(
177177
host_weights,
178178
weights_offsets,

fbgemm_gpu/codegen/training/backward/embedding_backward_split_host_template.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1116,7 +1116,7 @@ Tensor {{ bwd_mdesc }}_embedding_codegen_lookup_{{ optimizer }}_function(
11161116
{%- else %}
11171117
const c10::SymInt vbe_output_size = -1,
11181118
{%- endif %}
1119-
const bool mixed_D = true
1119+
const bool mixed_D = false
11201120
) {
11211121
// TODO: refactor into macro
11221122
{%- if has_gpu_support %}

0 commit comments

Comments
 (0)