We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 92ae286 commit 2e0f502Copy full SHA for 2e0f502
core/lowering/register_trt_placeholder_ops.cpp
@@ -20,7 +20,6 @@ RegisterOperators trt_placeholder_ops_reg({
20
[](Stack& stack) {
21
auto attn_mask = pop(stack).to<at::Tensor>();
22
if (attn_mask.scalar_type() == at::kBool) {
23
- attn_mask = attn_mask;
24
attn_mask.masked_fill_(attn_mask.logical_not(), -std::numeric_limits<float>::infinity());
25
}
26
return attn_mask;
cpp/src/types.cpp
@@ -158,6 +158,7 @@ TensorFormat::TensorFormat(at::MemoryFormat t) {
158
switch (t) {
159
case at::MemoryFormat::ChannelsLast:
160
value = TensorFormat::kChannelsLast;
161
+ break;
162
case at::MemoryFormat::Contiguous:
163
default:
164
value = TensorFormat::kContiguous;
0 commit comments