コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス 高性能並列固有値計算アルゴリズムの開発 櫻井鉄也1), 2), 二村保徳1),前田恭行1),矢野貴大1), Du Lei1), 2),今倉暁1),多田野寛人1) 1)筑波大学大学院システム情報工学研究科 2)科学技術振興機構CREST 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 研究課題 研究課題名 計算量子科学のニーズに対応した高性能固有値計算アルゴリズム の開発 体制 代表者 櫻井鉄也(筑波大学) 連携研究者 多田野寛人,今倉暁(筑波大学) 研究協力者 Du Lei,二村保徳,前田恭行,矢野貴大(筑波大学) 主な実施課題 階層型固有値解法によるバンド構造計算法の高性能化 確率的推定法を利用した状態密度計算法の開発 計算量子科学分野で現れるさまざまな固有値計算への展開 実問題での性能評価とボトルネックの改善 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 発表内容 はじめに 研究背景と目的 超並列固有値解析エンジン CRESTプロジェクトによるソフトウェア開発 周回積分による疎行列向け固有値解法 解法の概要と実行例 提案法が有効な固有値問題 おわりに 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 はじめに 研究背景と目的 固有値問題が現れるアプリケーションの高性能化 - 本課題では特に計算量子科学のニーズに対応 ペタスケール超規模のハードウェア性能を引き出す並列固有値 計算アルゴリズムの研究とその高性能実装技術を開発する アプリ A アプリ B アプリ C アプリ X 固有値解析エンジン アプリケーションを支える 基盤的技術 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 はじめに 大規模並列で性能を発揮する解法 数値例:高エネルギー加速器設計で現れる固有値問題(110万次元) - SS法*,Lanczos法ともに,内部で同じ線形ソルバーを使用 Cray XT4 (LBNL)における計算時間 時間(秒) 2500 提案法 (SS法*) 1000 従来法 (Shift-invert Lanczos法) 500 100 並列度が低いとき 8 32 並列度が高いとき 128 コア数 512 2048 *Sakurai, et al. 2003 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 CRESTプロジェクト 超並列固有値解析エンジンの開発 ポストペタスケール計算環境でのハードウェア性能を十分に引き出す 超並列固有値解析エンジンを開発 (H23年度〜H27年度) 主要メンバー - 櫻井(筑波大),張(名古屋大),今村(理研),山本(神戸大), 藏増(筑波大),星(鳥取大) アーキテクチャの超並列性を活用する階層型並列アルゴリズム 密行列・疎行列に対応 共通基盤技術としての線形計算手法・実装技術開発 高性能な線形計算手法,性能予測モデルによる評価, GPU/メニーコア向け実装 実問題での利用技術の開発 基礎科学・ナノ物質分野での実問題に適用し,高度利用技術を開発 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 周回積分を用いた固有値解法 基本アイディア:有理式に対する周回積分 Im Re 行列 A のレゾルベント : 固有値, : 固有ベクトル (ここでは簡単のため A は実対称とし,固有値は相異なるとする) 周回積分によるスペクトル射影 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 周回積分を用いた固有値解法 疎行列向け超並列固有値解法の概略 複素平面上の特定領域内部の固有ベクトル成分を抽出 周回積分は数値積分で近似 (各積分点では線形方程式を解く) 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 領域内固有値数の確率的推定法 パラメータの最適化のために確率的固有値数推定を用いる G 内の固有値数: 逆行列のトレースを以下のように近似する ここでベクトル の要素は 1 または -1 を等確率にとる GEP: Futamura, et al. (2010), NEP: Maeda et al. (2011) この推定値を利用して必要な部分空間サイズを決める 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 計算例 実空間密度汎関数法(RSDFT)によるバンド図計算 シリコンナノワイヤー9,924原子 - 行列サイズ:8,719,488次元,利用リソース:「京」 27,648コア - バンドギャップ付近の固有値を計算 Futamura, et al. (2013) 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 開発ソフトウェア 疎行列向けモジュール 固有値ソルバ,線形ソルバ,基本線形計算ライブラリ, アプリケーションインタフェイス 開発ソフトウェア MATLAB版:sseig.m - Webからダウンロード可 - 多項式,非線型版も利用可能 SLEPc版 (C言語):eps_ciss - SLEPc ver. 3.4 に実装 Fortran90版: - 京向けを近日公開予定 RSDFTのバンド図計算で利用 Trilinos版 (C++):開発中 Eigen-Supercomputing Sparse Engine Application Interface Eigensolver Block Linear Solver Base Modules: PETSc, Trilinos, MUMPS BLAS, SpBLAS, LAPACK, ScaLAPACK MatVec, Data Converter 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 GPU/メニーコアへの対応 密実対称行列の一般化固有値問題 10,000次元,最小固有値から10%の固有対を計算 BLAS/LAPACKルーチンとしてMAGMAを利用 miscの大部分(SVD, Rayleigh-Ritz)は,現状では単一GPU or CPU実装 Elapsed time [sec] GPUクラスタ HA-PACS (Tsukuba)での実験 180 160 140 120 100 80 60 40 20 0 misc Solve LU 1GPU 2GPU 4GPU 8GPU 16GPU Yano, et al. accepted 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 提案法が有効な問題 固有値を求める領域 Im Im Re Re 指定した範囲 複素平面上の指定領域 リング状領域 求めたい範囲やその範囲内にある固有値数が ある程度予想できる問題に対して効果が高い 固有値問題の種類 標準固有値問題 一般化固有値問題 多項式固有値問題 非線形固有値問題 (ex. 平方根,有理式,etc.) 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 おわりに 大規模並列で性能を発揮する解法の開発 高性能固有値解析ソフトウェアを開発 京版は近日中に公開予定,今後はポストペタへ アルゴリズムレベルでの階層構造 周回積分により固有成分をフィルタリング 独立な複数の線形方程式に帰着 実アプリケーション RSDFTのバンド図計算 今後の課題 実用性を考慮したアプリケーション向けインターフェイスの開発 計算量子科学分野で現れるさまざまな固有値計算への展開 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」 References 1) Y. Futamura, T. Sakurai, S. Furuya and J.-I. Iwata, Proc. VECPAR 2012, LNCS 7851, 226–235 (2013). 2) T. Mizusaki, K. Kaneko, M. Honma, and T. Sakurai, Phys. Rev. C 82, 024310 (2010). 3) Y. Nagai, Y. Shinohara, Y. Futamura, Y. Ota and T. Sakurai, arXiv:1303.3683 (2013). 4) H. Ohno, Y. Kuramashi, H. Tadano and T. Sakurai, JSIAM Letters 2, 115-118 (2010). 5) T. Sakurai and H. Sugiura, J. Comput. Appl. Math. 159, 119-128 (2003). 6) T. Sakurai, H. Tadano, T. Ikegami and U. Nagashima, Taiwanese J. Math. 14, 855-867 (2010). 7) T. Sakurai, H. Tadano and Y. Futamura, J. Alg. Comput. Tech. 7 (in print). 8) T. Yano, Y. Futamura and T. Sakurai, Proc. 3PGCIC2013 (accepted). 新学術領域「コンピューティクスによる物質デザイン:複合相関と非平衡ダイナミクス」
© Copyright 2024 ExpyDoc