Add is_compiling() checks instead of disabling extensions during torch_geometric.compile

Add is_compiling() checks instead of disabling extensions during torch_geometric.compile #16572