This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
SYCL_PI_TRACE[basic]: Plugin found and successfully loaded: libpi_opencl.so [ PluginVersion: 12.27.1 ] | |
---> piPlatformsGet( | |
<unknown> : 0 | |
<nullptr> | |
<unknown> : 0x7fff562f834c | |
) ---> pi_result : PI_SUCCESS | |
---> piPlatformsGet( | |
<unknown> : 3 | |
<unknown> : 0x55ec664cee70 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import torch | |
import math | |
import numpy as np | |
def kmeans_single( | |
# NB: best performance might depend on the layout for `X` and `centroids` | |
# TODO: benchmark and warns or error out if the layout is not adapted | |
X, # (n_samples, n_features) | |
sample_weight, # (n_samples,) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import math | |
import numpy as np | |
import torch | |
def kneighbors( | |
# NB: best performance might depend on the layout for `X` and `centroids` | |
# TODO: benchmark and warns or error out if the layout is not adapted | |
query, # (n_queries, n_features) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
6.354323008901774372e-01,7.721565844991642447e-01 | |
9.988797155850336340e-01,-4.732138832243185100e-02 | |
-8.296770135526191359e-01,5.582437220268644529e-01 | |
6.870527672236669270e-01,7.266075247685656047e-01 | |
-3.041148323275185694e-01,9.526353808033822812e-01 | |
-2.877167672261670139e-01,-9.577155432887802800e-01 | |
-9.805753940631429799e-01,1.961425414281967594e-01 | |
9.326545075388225081e-01,-3.607707992167831224e-01 | |
8.717063187093219145e-01,-4.900286664290592342e-01 | |
7.221165966101161793e-01,6.917713646142219508e-01 |