File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
fbgemm_gpu/codegen/training/backward Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -159,7 +159,7 @@ Tensor {{ mdesc }}_embedding{{ ndesc }}_backward_codegen_{{ optimizer }}_{{ desc
159
159
{%- endif %}
160
160
161
161
// short-circuit if there are zero indices.
162
- if (TORCH_GUARD_SIZE_OBLIVIOUS (indices.sym_numel ().sym_eq (0 ))) {
162
+ if (TORCH_GUARD_OR_FALSE (indices.sym_numel ().sym_eq (0 ))) {
163
163
{%- if dense %}
164
164
return grad_dev_weights;
165
165
{%- elif optimizer == " none" %}
@@ -213,7 +213,7 @@ Tensor {{ mdesc }}_embedding{{ ndesc }}_backward_codegen_{{ optimizer }}_{{ desc
213
213
214
214
// Took allocation from https://www.internalfb.com/code/fbsource/fbcode/deeplearning/fbgemm/fbgemm_gpu/src/split_embeddings_utils.cu?lines=339-347
215
215
Tensor sorted_linear_indices_run;
216
- if (TORCH_GUARD_SIZE_OBLIVIOUS (total_unique_indices.sym_gt (0 ))) {
216
+ if (TORCH_GUARD_OR_TRUE (total_unique_indices.sym_gt (0 ))) {
217
217
sorted_linear_indices_run = at::empty_symint ({total_unique_indices}, indices.options ());
218
218
} else {
219
219
sorted_linear_indices_run = at::empty_like (indices);
You can’t perform that action at this time.
0 commit comments