Add `can_torch_compile`

torch-ext/backward_marker_test/layers.py

@@ -4,11 +4,14 @@ from torch.nn import functional as F
 
 
 class LinearImplicitBackward(nn.Module):
+    can_torch_compile = True
+
     def forward(self, input: torch.Tensor) -> torch.Tensor:
         return F.linear(input, self.weight, self.bias)
 
 
 class LinearBackward(nn.Module):
+    can_torch_compile = True
     has_backward = True
 
     def forward(self, input: torch.Tensor) -> torch.Tensor:
@@ -16,6 +19,7 @@ class LinearBackward(nn.Module):
 
 
 class LinearNoBackward(nn.Module):
+    can_torch_compile = True
     has_backward = False
 
     def forward(self, input: torch.Tensor) -> torch.Tensor: