Skip to content

Commit 0b90c7e

Browse files
committed
cleanup
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 898634c commit 0b90c7e

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

third_party/intel/backend/driver.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -71,26 +71,18 @@ class CompilationHelper:
7171
def __init__(self):
7272
self._library_dir = None
7373
self._include_dir = None
74-
self.libraries = ['ze_loader', 'sycl', 'torch']
74+
self.libraries = ['ze_loader', 'sycl']
7575

7676
@cached_property
7777
def _compute_compilation_options_lazy(self):
78-
import torch
7978
ze_root = os.getenv("ZE_PATH", default="/usr/local")
8079
include_dir = [os.path.join(ze_root, "include")]
8180

8281
include_dir, library_dir = find_sycl(include_dir)
8382

8483
dirname = os.path.dirname(os.path.realpath(__file__))
8584
include_dir += [os.path.join(dirname, "include")]
86-
include_dir += [
87-
os.path.join(torch.utils.cmake_prefix_path, "../../include"),
88-
os.path.join(torch.utils.cmake_prefix_path, "../../include/torch/csrc/api/include"),
89-
]
9085
library_dir += [os.path.join(dirname, "lib")]
91-
library_dir += [
92-
os.path.join(torch.utils.cmake_prefix_path, "../../lib"),
93-
]
9486

9587
self._library_dir = library_dir
9688
self._include_dir = include_dir
@@ -226,7 +218,6 @@ def format_of(ty):
226218
#include <iomanip>
227219
#include <level_zero/ze_api.h>
228220
#include <sycl/sycl.hpp>
229-
#include <ATen/record_function.h>
230221
231222
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
232223
#include <Python.h>
@@ -319,7 +310,6 @@ def format_of(ty):
319310
static void sycl_kernel_launch(uint32_t gridX, uint32_t gridY, uint32_t gridZ, int num_warps, int threads_per_warp, int shared_memory, sycl::queue& stream, sycl::kernel& kernel_ptr {', ' + arg_decls if len(arg_decls) > 0 else ''}) {{
320311
321312
std::string kernel_name = kernel_ptr.get_info<sycl::info::kernel::function_name>();
322-
RECORD_FUNCTION("XPU Triton kernel: " + kernel_name, {{}});
323313
void *params[] = {{ {', '.join(f"&arg{i}" for i in signature.keys() if i not in constants)} }};
324314
uint32_t num_params = sizeof(params)/sizeof(params[0]);
325315
uint32_t expected_num_params = kernel_ptr.get_info<sycl::info::kernel::num_args>();

0 commit comments

Comments
 (0)