derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
107 lines
2.1 KiB
Matlab
107 lines
2.1 KiB
Matlab
% ********************************************************************
|
|
% Copyright (C) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
|
% ********************************************************************
|
|
|
|
function [T,Q] = print_prob( A, dir_in )
|
|
%
|
|
% [T,Q] = print_prob( A [, dir] )
|
|
%
|
|
%
|
|
% Given a sparse matrix, write out the permutations and factorization
|
|
%
|
|
% The files are created in the current local directory by default
|
|
% If optional "dir" is given, the files are created under that directory
|
|
|
|
dir = '.';
|
|
if (nargin >= 2),
|
|
dir = dir_in;
|
|
is_absolute_dir = (dir(1) == '/');
|
|
if (is_absolute_dir),
|
|
[status,msg,msgid] = mkdir('',dir);
|
|
else
|
|
[status,msg,msgid] = mkdir('.',dir);
|
|
end;
|
|
isok = (status == 1);
|
|
if (~isok),
|
|
error(sprintf('print_prob: mkdir for %s return msg=%s, msgid=%g', ...
|
|
dir, msg, msgid ));
|
|
return;
|
|
end;
|
|
end;
|
|
|
|
has_final_slash = (dir( numel(dir) ) == '/');
|
|
if (~has_final_slash),
|
|
dir = strcat(dir,'/');
|
|
end;
|
|
|
|
nrow = size(A,1);
|
|
ncol = size(A,2);
|
|
[Ap,Ai,Ax] = sparse2csr( A );
|
|
print_gvec(strcat(dir,'ptrA'),Ap);
|
|
print_gvec(strcat(dir,'indA'),Ai);
|
|
print_gvec(strcat(dir,'valA'),Ax);
|
|
|
|
|
|
|
|
m = 30;
|
|
X = reshape( 1:(ncol*m), ncol, m );
|
|
B = A * X;
|
|
|
|
print_gvec(strcat(dir,'X_30'),X(:,1:30));
|
|
print_gvec(strcat(dir,'X_10'),X(:,1:10));
|
|
print_gvec(strcat(dir,'X_1'),X(:,1:1));
|
|
|
|
print_gvec(strcat(dir,'B_30'),B(:,1:30));
|
|
print_gvec(strcat(dir,'B_10'),B(:,1:10));
|
|
print_gvec(strcat(dir,'B_1'),B(:,1:1));
|
|
|
|
|
|
[L, is_indefinite, Q] = chol(A,'lower');
|
|
if (is_indefinite),
|
|
|
|
% perform LU instead
|
|
|
|
[L,U,P,Q] = lu(A);
|
|
|
|
np = size(P,2);
|
|
pvec = P*reshape(1:np,np,1);
|
|
pvec = pvec - 1;
|
|
print_gvec(strcat(dir,'P'),pvec);
|
|
|
|
nq = size(Q,2);
|
|
qvec = Q * reshape(1:nq,nq,1);
|
|
qvec = qvec - 1;
|
|
print_gvec(strcat(dir,'Q'),qvec);
|
|
|
|
|
|
T = tril(L,-1) + triu(U);
|
|
|
|
|
|
else
|
|
|
|
|
|
tiny = 1e-100;
|
|
T = L + triu(L',1)*tiny + triu(Q'*A*Q,1);
|
|
|
|
nq = size(Q,2);
|
|
qvec = Q * reshape(1:nq,nq,1);
|
|
qvec = qvec - 1;
|
|
print_gvec(strcat(dir,'Q'),qvec);
|
|
|
|
end
|
|
|
|
|
|
|
|
[Tp,Ti,Tx] = sparse2csr( T );
|
|
|
|
|
|
|
|
|
|
print_gvec(strcat(dir,'ptrT'),Tp);
|
|
print_gvec(strcat(dir,'indT'),Ti);
|
|
print_gvec(strcat(dir,'valT'),Tx);
|
|
|
|
|
|
|
|
end
|