frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix vscode test discovery (#107404)

Browse Source 
Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/107404
Approved by: https://github.com/wconstab
This commit is contained in:
Michael Lazos
2023-08-18 03:50:46 +00:00
committed by PyTorch MergeBot
parent 6cb0128c8a
commit 2bb59a9ac6
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test/inductor/test_extension_backend.py
View File

@ -7,10 +7,18 @@ import unittest
import torch
import torch._dynamo
import torch.utils.cpp_extension
from extension_backends.extension_codegen_backend import (
ExtensionScheduling,
ExtensionWrapperCodegen,
)
try:
from extension_backends.extension_codegen_backend import (
ExtensionScheduling,
ExtensionWrapperCodegen,
)
except ImportError:
from .extension_backends.extension_codegen_backend import (
ExtensionScheduling,
ExtensionWrapperCodegen,
)
from torch._C import FileCheck
from torch._inductor import metrics
from torch._inductor.codegen.common import (