remove patch code

AlibabaPAI · Dec 13, 2024 · 3ecf7e4 · 3ecf7e4
1 parent 4176f1c
commit 3ecf7e4
Show file tree

Hide file tree

Showing 8 changed files with 0 additions and 13 deletions.
diff --git a/benchmarks/accuracy_benchmark/README.md → benchmarks/accuracy/README.md b/benchmarks/accuracy_benchmark/README.md → benchmarks/accuracy/README.md
diff --git a/benchmarks/accuracy_benchmark/fastchat.sh → benchmarks/accuracy/fastchat.sh b/benchmarks/accuracy_benchmark/fastchat.sh → benchmarks/accuracy/fastchat.sh
diff --git a/benchmarks/accuracy_benchmark/llama.sh → benchmarks/accuracy/llama.sh b/benchmarks/accuracy_benchmark/llama.sh → benchmarks/accuracy/llama.sh
diff --git a/...ks/accuracy_benchmark/llama_fsdp_acc.json → benchmarks/accuracy/llama_fsdp_acc.json b/...ks/accuracy_benchmark/llama_fsdp_acc.json → benchmarks/accuracy/llama_fsdp_acc.json
diff --git a/.../accuracy_benchmark/llama_fsdp_torch.json → benchmarks/accuracy/llama_fsdp_torch.json b/.../accuracy_benchmark/llama_fsdp_torch.json → benchmarks/accuracy/llama_fsdp_torch.json
diff --git a/benchmarks/accuracy_benchmark/run.sh → benchmarks/accuracy/run.sh b/benchmarks/accuracy_benchmark/run.sh → benchmarks/accuracy/run.sh
diff --git a/benchmarks/accuracy_benchmark/run_clm.py → benchmarks/accuracy/run_clm.py b/benchmarks/accuracy_benchmark/run_clm.py → benchmarks/accuracy/run_clm.py
diff --git a/torchacc/__init__.py b/torchacc/__init__.py
@@ -136,16 +136,3 @@ def _set_env():
 patch.patch_autocast()
 decompose.replace_decompose()
 _set_env()
-
-import torch
-
-original_init = torch.autocast.__init__
-
-
-def patched_init(self, device_type: str, *args, **kwargs):
-    if device_type == 'xla':
-        device_type = 'cuda'
-    original_init(self, device_type, *args, **kwargs)
-
-
-torch.autocast.__init__ = patched_init