fix set item in tensor

frontend/guard_tracker.py

@@ -173,7 +173,13 @@ def get_common_device(arg: Any) -> None:
 
         def as_fx_node(arg: Any) -> NodeArgs:
             if isinstance(arg, (tuple, list)):
-                return fx_immutable.immutable_list([as_fx_node(x) for x in arg])
+                if isinstance(arg, list):
+                    return fx_immutable.immutable_list(
+                        [as_fx_node(x) for x in arg])
+                else:
+                    return tuple(
+                        fx_immutable.immutable_list(
+                            [as_fx_node(x) for x in arg]))
             if isinstance(arg, slice):
                 return slice(as_fx_node(arg.start), as_fx_node(arg.stop),
                              as_fx_node(arg.step))
@@ -1735,6 +1741,12 @@ def set_if_inplace_return() -> None:
                        "__name__") and func.__name__ in ("flatten_parameters",
                                                          "numel", "children"):
                 return
+            if hasattr(func, "__module__"
+                      ) and func.__module__ == 'torch.autograd.profiler':
+                return
+            elif hasattr(func, "__self__") and isinstance(
+                    func.__self__, torch.autograd.profiler.record_function):
+                return
             print("record function in graph", func)
             self.state.record_function(
                 func,

test/test_tensor.py

@@ -420,3 +420,20 @@ def test_no_grad(caplog):
         compiled = compile(run_no_grad)
         run_and_check(compiled, [MISS], 1, caplog, expect, inp)
         run_and_check(compiled, [HIT], 1, caplog, expect, inp)
+
+
+def tensor_set_item(x):
+    length = x.shape[0]
+    for i in range(length):
+        x[i, i, i] = 1.0
+    return x
+
+
+def test_tensor_set_item(caplog):
+    reset()
+    with torch.no_grad():
+        input = torch.rand([4, 4, 4, 4])
+        expect = tensor_set_item(input)
+        compiled = compile(tensor_set_item)
+        run_and_check(compiled, [MISS], 1, caplog, expect, input)
+        run_and_check(compiled, [HIT], 1, caplog, expect, input)