@@ -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