Illegal instruction (core dumped)

[JacopoRaffi]

Hi, I am trying to run the example in pytorch_geometric/examples/distributed/pyg/, when i execute the script python partition_graph.py --dataset=ogbn-products --root_dir=../../../data --num_partitions=2 it gives me the following error: Illegal instruction (core dumped). I found out that the code that gives the error is

if cluster is None and torch_geometric.typing.WITH_METIS:
            cluster = pyg_lib.partition.metis(
                indptr.cpu(),
                index.cpu(),
                self.num_parts,
                recursive=self.recursive,
            ).to(edge_index.device)
        if cluster is None:
            raise ImportError(f"'{self.__class__.__name__}' requires either "
                              f"'pyg-lib' or 'torch-sparse'")

in particular the function pyg_lib.partition.metis seems to give the error.
I am using:
pyg-lib 0.3.1+pt21cpu
torch 2.1.2
torch_geometric 2.5.3.
Thank you in advance for all the answers.