search

intel / torch-xpu-ops

Apache License 2.0
30 stars 21 forks source link

[Windows][PT2.6] Split larger Other Kernels lib #1096

Open ratnampa opened 3 days ago

ratnampa commented 3 days ago

torch_xpu_ops_sycl_kernels leads to around 1.83GB in size on windows, splitting it to reduce the lib size.

New libs introduced in this PR:

torch_xpu_ops_sycl_tensor_srcs torch_xpu_ops_sycl_norm_loss_srcs torch_xpu_ops_sycl_poly_srcs torch_xpu_ops_sycl_dist_srcs

fengyuan14 commented 2 days ago

Could you also add xe2-hpg to Windows AOT list in the PR?

ratnampa commented 2 days ago

Could you also add xe2-hpg to Windows AOT list in the PR?

Sure, will add this one itself.

fengyuan14 commented 1 day ago

LGTM, @ratnampa Could you list the sizes of all kernel libraries in the PR description?