JP2024028901A - ハードウェアにおけるスパース行列乗算 - Google Patents

ハードウェアにおけるスパース行列乗算 Download PDF

Info

Publication number
JP2024028901A
JP2024028901A JP2023206881A JP2023206881A JP2024028901A JP 2024028901 A JP2024028901 A JP 2024028901A JP 2023206881 A JP2023206881 A JP 2023206881A JP 2023206881 A JP2023206881 A JP 2023206881A JP 2024028901 A JP2024028901 A JP 2024028901A
Authority
JP
Japan
Prior art keywords
shard
sparse
input
vector
matrix
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Pending
Application number
JP2023206881A
Other languages
English (en)
Inventor
レイナー・アルウィン・ポープ
Alwyn Pope Reiner
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Google LLC
Original Assignee
Google LLC
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Google LLC filed Critical Google LLC
Publication of JP2024028901A publication Critical patent/JP2024028901A/ja
Pending legal-status Critical Current

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F17/00Digital computing or data processing equipment or methods, specially adapted for specific functions
    • G06F17/10Complex mathematical operations
    • G06F17/16Matrix or vector computation, e.g. matrix-matrix or matrix-vector multiplication, matrix factorization
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F15/00Digital computers in general; Data processing equipment in general
    • G06F15/76Architectures of general purpose stored program computers
    • G06F15/80Architectures of general purpose stored program computers comprising an array of processing units with common control, e.g. single instruction multiple data processors
    • G06F15/8046Systolic arrays
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F7/00Methods or arrangements for processing data by operating upon the order or content of the data handled
    • G06F7/38Methods or arrangements for performing computations using exclusively denominational number representation, e.g. using binary, ternary, decimal representation
    • G06F7/48Methods or arrangements for performing computations using exclusively denominational number representation, e.g. using binary, ternary, decimal representation using non-contact-making devices, e.g. tube, solid state device; using unspecified devices
    • G06F7/483Computations with numbers represented by a non-linear combination of denominational numbers, e.g. rational numbers, logarithmic number system or floating-point numbers
    • G06F7/487Multiplying; Dividing
    • G06F7/4876Multiplying
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F7/00Methods or arrangements for processing data by operating upon the order or content of the data handled
    • G06F7/38Methods or arrangements for performing computations using exclusively denominational number representation, e.g. using binary, ternary, decimal representation
    • G06F7/48Methods or arrangements for performing computations using exclusively denominational number representation, e.g. using binary, ternary, decimal representation using non-contact-making devices, e.g. tube, solid state device; using unspecified devices
    • G06F7/50Adding; Subtracting
    • G06F7/501Half or full adders, i.e. basic adder cells for one denomination
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F7/00Methods or arrangements for processing data by operating upon the order or content of the data handled
    • G06F7/76Arrangements for rearranging, permuting or selecting data according to predetermined rules, independently of the content of the data
    • G06F7/78Arrangements for rearranging, permuting or selecting data according to predetermined rules, independently of the content of the data for changing the order of data flow, e.g. matrix transposition or LIFO buffers; Overflow or underflow handling therefor
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N20/00Machine learning
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods

Landscapes

  • Engineering & Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Theoretical Computer Science (AREA)
  • Computational Mathematics (AREA)
  • Mathematical Analysis (AREA)
  • Pure & Applied Mathematics (AREA)
  • Computing Systems (AREA)
  • Mathematical Physics (AREA)
  • Mathematical Optimization (AREA)
  • General Engineering & Computer Science (AREA)
  • Data Mining & Analysis (AREA)
  • Software Systems (AREA)
  • Databases & Information Systems (AREA)
  • Algebra (AREA)
  • Artificial Intelligence (AREA)
  • Evolutionary Computation (AREA)
  • Computer Hardware Design (AREA)
  • Nonlinear Science (AREA)
  • Biophysics (AREA)
  • Biomedical Technology (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Health & Medical Sciences (AREA)
  • Molecular Biology (AREA)
  • General Health & Medical Sciences (AREA)
  • Computational Linguistics (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Medical Informatics (AREA)
  • Complex Calculations (AREA)
  • Neurology (AREA)

Abstract

【課題】予め定められた次元閾値までの任意のサイズの入力部分行列を効率的に処理し、対応する長さのベクトルとして積を入力に出力することができるスパース行列乗算のためのスパースシャードシステム、方法及びコンピュータ読取可能記憶媒体を含む装置を提供する。【解決手段】スパースシャード行列乗算システムは、スパースシャード101A~101Pのアレイ101を含む。各スパースシャードは、入力部分行列及び入力部分ベクトルを受け取る。入力部分行列は、予め定められた最大非ゼロ閾値以下の数の非ゼロ値を有する。スパースシャードは、複数の乗算回路によって、ベクトル値に入力部分行列のそれぞれの非ゼロ値を乗算した1つ以上の積を計算することができる。スパースシャードは、スパースシャードへの出力として、1つ以上の積を用いて、シャード入力ベクトルをシャード入力行列に適用した積であるシャード出力ベクトルを生成することができる。【選択図】図1

Description

背景
スパース行列は、行列の要素として非ゼロ値よりもゼロ値の方が割合が高い行列である。異なるスパース行列は、ゼロ値と非ゼロ値との割合に基づいてさまざまな程度のスパース性を有し得る。非ゼロ値よりもゼロ値の方が割合が高い行列は、非ゼロ値よりもゼロ値の方が割合が低い行列よりもスパース性が高いと言われる。
ニューラルネットワークは、受けた入力に対する出力を予測するための1つ以上の非線形演算層を含む機械学習モデルである。入力層および出力層に加えて、1つ以上の隠れ層を含むニューラルネットワークもある。各隠れ層の出力は、ニューラルネットワークの別の隠れ層または出力層に入力され得る。ニューラルネットワークの各層は、当該層の1つ以上のモデルパラメータの値に従って、受けた入力からそれぞれの出力を生成することができる。モデルパラメータは、ニューラルネットワークが正確な出力を生成するように訓練アルゴリズムを通して求められる重みまたはバイアスであってもよい。ニューラルネットワークの層のモデルパラメータ値は、行列またはテンソルの要素として表すことができる。
簡単な概要
本開示の局面は、ハードウェアにおけるスパース行列密ベクトル乗算に関する。
本開示の一局面は、複数の乗算回路を含むスパースシャードを含むシステムを提供し、上記スパースシャードは、予め定められた最大非ゼロ閾値以下の数の非ゼロ値を含むシャード入力行列を受け、複数のベクトル値を含むシャード入力ベクトルを受け、上記乗算回路の各々について、上記シャード入力行列のそれぞれの非ゼロ値を受け、上記複数の乗算回路によって、ベクトル値に上記シャード入力行列の上記それぞれの非ゼロ値を乗算した1つ以上の積を生成し、上記スパースシャードへの出力として、上記1つ以上の積を用いて、上記シャード入力ベクトルを上記シャード入力行列に適用した積であるシャード出力ベクトルを生成するように構成されている。
本開示の別の局面は、複数のスパースシャードを含むシステムによって実行されると上記システムに動作を実行させる命令を格納した1つ以上の非一時的なコンピュータ読取可能記憶媒体を提供し、上記動作は、複数の乗算回路を含むスパースシャードによって、予め定められた最大非ゼロ閾値以下の数の非ゼロ値を含むシャード入力行列と、複数のベクトル値を含むシャード入力ベクトルとを受けることと、上記乗算回路の各々について、上記シャード入力行列のそれぞれの非ゼロ値を受けることと、上記スパースシャードの上記複数の乗算回路によって、それぞれのベクトル値に上記それぞれの非ゼロ値を乗算した1つ以上の積を生成することと、上記スパースシャードへの出力として、上記1つ以上の積を用いて、上記シャード入力ベクトルを上記シャード入力行列に適用した積であるシャード出力ベクトルを生成することとを含む。
本開示の別の局面は、方法を提供し、上記方法は、複数の乗算回路を含むスパースシャードが、予め定められた最大非ゼロ閾値以下の数の非ゼロ値を含むシャード入力行列と、複数のベクトル値を含むシャード入力ベクトルとを受けることと、上記乗算回路の各々について、上記シャード入力行列の上記非ゼロ値のそれぞれを受けることと、上記スパースシャードの上記複数の乗算回路が、それぞれのベクトル値に上記シャード入力行列の上記
非ゼロ値のそれぞれを乗算した1つ以上の積を生成することと、上記スパースシャードが、上記スパースシャードへの出力として、上記1つ以上の積を用いて、上記シャード入力ベクトルを上記シャード入力行列に適用した積であるシャード出力ベクトルを生成することとを含む。
上述のおよび他の局面の各々は、任意に以下の特徴のうちの1つ以上を単独でまたは組み合わせて含み得る。一実現例は、以下の特徴のすべてを組み合わせて含み得る。
上記シャード出力ベクトルの長さは1よりも大きい。
上記スパースシャードは複数のスパースシャードのうちの1つであり、上記複数のスパースシャードは、システム入力行列の部分行列である複数のシャード入力行列を受け、システム入力ベクトルの部分ベクトルである複数のシャード入力ベクトルを受け、上記複数のスパースシャードによって、上記システム入力ベクトルを上記システム入力行列に適用した積を表すシステム出力ベクトルを生成するように構成されている。
上記複数のスパースシャードはシストリックアレイとして配列され、上記シストリックアレイは、上記シストリックアレイの列次元に沿ったスパースシャードの1つ以上のグループを含み、上記システム出力ベクトルを生成するために、上記1つ以上のプロセッサはさらに、上記シストリックアレイの上記列次元に沿ったグループごとに、上記グループ内の各スパースシャードのそれぞれのシャード出力ベクトルを合算してそれぞれの列出力ベクトルを生成し、各グループの上記それぞれの列出力ベクトルを連結して上記システム出力ベクトルを生成するように構成されている。
各乗算回路は、上記スパースシャードについての上記シャード入力行列のそれぞれからの上記それぞれの非ゼロ値を含むそれぞれのレジスタに結合される。
上記複数の乗算回路における乗算回路の数は、上記予め定められた最大非ゼロ閾値と等しい。
上記スパースシャードはクロスバー回路をさらに含み、上記スパースシャードはさらに、上記クロスバー回路によって、上記シャード入力ベクトルの上記複数のベクトル値を受け、上記複数の乗算回路の各々への入力として、上記クロスバー回路によって、上記複数のベクトル値のうちの1つのベクトル値を送るように構成されている。
上記スパースシャードはさらに、上記シャード入力行列の同じ列の非ゼロ値を、上記複数の乗算回路のうちの隣接する乗算回路のレジスタにロードするように構成されている。
上記スパースシャードはさらに、上記シャード入力行列の各列に沿った非ゼロ値の位置を少なくとも指定する1つ以上の制御値を受けるように構成されており、上記スパースシャードの上記クロスバー回路はさらに、上記1つ以上の制御値を受け、上記1つ以上の制御値に従って、上記シャード入力行列の同じ列に沿った非ゼロ値が乗算されるベクトル値を、隣接する乗算回路に送るように構成されている。
上記スパースシャードは複数の加算回路をさらに含み、上記スパースシャードは1つ以上のセグメントマーカをさらに含み、各セグメントマーカは、上記セグメントマーカにロードされたそれぞれの制御値の値に基づいて、上記複数の加算回路のうちのそれぞれの加算回路への入力をゲート開閉するように構成されており、上記スパースシャードはさらに、上記1つ以上の制御値の少なくとも一部を上記1つ以上のセグメントマーカにロードするように構成されており、上記シャード入力行列の1列目の非ゼロ値についての加算回路は、上記1列目とは異なる上記シャード入力行列の2列目の非ゼロ値を含む隣接する加算
回路の入力を受けないようにゲート開閉されており、上記スパースシャードはさらに、上記複数の加算回路によって、上記1つ以上の積の1つ以上の合計を生成するように構成されており、上記1つ以上の合計の各々は、上記シャード入力行列の列の1つ以上の非ゼロ値に上記シャード入力ベクトルの1つ以上のそれぞれの値を乗算したそれぞれのセグメント化合計である。
上記複数の加算回路は並列セグメント化合計回路を形成し、1つ以上の上記セグメント化合計の各々は、セグメントマーカによってゲート開閉されていない隣接する加算回路への出力の合計である。
上記クロスバー回路は第1のクロスバー回路であり、上記スパースシャードは第2のクロスバー回路をさらに含み、上記第2のクロスバー回路は、1つ以上の上記セグメント化合計を受け、上記1つ以上の制御値に従って1つ以上の上記セグメント化合計を配列して、上記スパースシャードついて上記シャード出力ベクトルのそれぞれを生成するように構成されている。
上記第2のクロスバー回路はベネシュネットワークを形成してもよく、上記シャード入力行列は正方行列である。
本開示の別の局面は、1つ以上のプロセッサと、上記1つ以上のプロセッサによって実行されると上記1つ以上のプロセッサに動作を実行させる命令を格納した1つ以上のメモリデバイスとを含むシステムを提供し、上記動作は、ゼロ値および非ゼロ値を含む入力行列を受けることと、上記入力行列を複数の部分行列に区分することとを含み、各部分行列の非ゼロ値の数は予め定められた最大非ゼロ閾値以下であり、各部分行列の次元は予め定められた次元閾値以下である。
上記動作はさらに、部分行列ごとに、上記部分行列の各列に沿った非ゼロ値の位置を指定する1つ以上のそれぞれの制御値を生成することを含み得る。
上記動作はさらに、各部分行列および各部分行列についての上記1つ以上のそれぞれの制御値を、各部分行列および上記部分行列についてのそれぞれの制御値を処理するように構成された複数のスパースシャードに送ることを含み得る。上記複数のスパースシャードの各々は、部分行列と、上記部分行列についての上記1つ以上のそれぞれの制御値と、入力ベクトルの少なくとも一部とを受け、上記部分行列と上記入力ベクトルの上記一部との積を表すそれぞれの出力シャードベクトルを生成するように構成されている。請求項21に記載のシステムであって、上記入力行列を区分することは、上記行列を、上記複数のスパースシャードにおけるスパースシャードの数と等しい数の部分行列に区分することを含む。
本開示の他の局面は、対応するシステム、装置、および1つ以上の非一時的なコンピュータ読取可能記憶媒体に格納されたコンピュータプログラムを含む。
本開示の局面に係る、スパースシャードのアレイを含むシステムの一例のブロック図である。 本開示の局面に係る、スパースシャードの一例のブロック図である。 スパースシャードのシャード入力行列の一例を示す図である。 シャード入力行列の一例の非ゼロ値のベクトルを示す図である。 シャード入力行列の一例についての制御値のベクトルを示す図である。 シャード入力行列と、シャード入力ベクトルと、シャード入力行列についての制御値とを受けるスパースシャードによる行列乗算の一例を示す図である。 本開示の局面に係る、スパースシャード上のスパース行列の部分行列にシステム入力ベクトルを乗算するためのプロセスの一例のフロー図である。 複数のスパースシャードからのシステム入力ベクトルとシステム入力行列との積を表すシステム出力ベクトルを生成するプロセスの一例のフロー図である。 システム入力行列の一例およびシステム入力ベクトルを示す図である。 システム入力行列およびシステム入力ベクトルの区分を示す図である。 区分されたシステム入力行列、およびシステム入力行列とシステム入力ベクトルとを乗算した積を表すシステム出力ベクトルを示す図である。 本開示の局面に係る、シャード入力行列の1つ以上の制御値を用いてスパースシャードを構成するためのプロセスの一例のフロー図である。 図8Aのプロセスに従って構成されたスパースシャードを用いて行列-ベクトル乗算を実行するためのプロセスの一例のフロー図である。 本開示の局面に係る、システム入力行列から部分行列を生成するためのプロセスの一例のフロー図である。 本開示の局面に係る、スパース行列乗算システムを実現するコンピューティング環境の一例のブロック図である。
詳細な説明
概要
本開示の局面は、スパース行列密ベクトル乗算のために構成された1つ以上の集積回路を含むシステムに関する。複数のスパースシャード(sparse shard)のシステムは、スパースシャードごとに、入力システム行列およびベクトルの部分行列および部分ベクトルを受けることができる。各スパースシャードは、集積回路の少なくとも一部であってもよく、乗算回路および加算回路などの複数の演算ユニットを実現することができる。各スパースシャードは、最大非ゼロ閾値以下の数の非ゼロ値を有する部分行列を受けるように構成されており、この最大非ゼロ閾値は、システムが、たとえば本明細書に記載されているようにスパースシャードと任意に1つ以上の他のコンポーネントとを含むチップとして実現される場合に、予め定めることができる。
各スパースシャードはさらに、本明細書に記載されている制御値などのメタデータを受けることができ、スパースシャードはこのメタデータを用いて、異なる入力を各乗算回路または加算回路に導くことができ、あるユニットからの出力が別のユニットに渡されるときにゲート開閉することができる。受けたメタデータに従ってスパースシャードを構成することによって、スパースシャードは、予め定められた次元閾値までの任意のサイズの入力部分行列を効率的に処理し、対応する長さのベクトルとして積を入力に出力することができる。
部分行列を受ける一部として、各スパースシャードは、部分行列の各列における非ゼロ値の位置を表す1つ以上の制御値を受けることができる。この1つ以上の制御値を用いて、スパースシャードは、スパースシャードが部分行列と部分ベクトルとの積を表すシャード出力ベクトルを生成するように、部分行列および部分ベクトルの個々の値をどのように乗算、加算、および配列すべきかを調整するように構成され得る。システムはさらに、たとえばシステム入力ベクトルにシステム入力行列を乗算することによって、各シャードのシャード出力ベクトルから、システム入力ベクトルをシステム入力行列に適用した積を表すシステム出力ベクトルを生成するように構成され得る。
また、本開示の局面は、スパースシャードのアレイによる処理のためにスパース行列を前処理するためのシステムに関する。1つ以上のプロセッサのシステムは、入力行列を複
数の部分行列に区分するように構成され得、各部分行列の次元は、スパースシャードの最大行列入力サイズを指定する予め定められた次元閾値以下である。区分の一部として、システムは、部分行列のうちのいずれかが、予め定められた最大非ゼロ閾値よりも大きい数の非ゼロ値を含むか否かを識別し、それに応じて、識別した部分行列と同じ行または列に沿って部分行列を再区分することができる。システムは、部分行列の次元が次元閾値内にあり、部分行列の非ゼロ値の数が予め定められた非ゼロ閾値以下であるように、スパースシャードごとに部分行列を生成するまでこのプロセスを繰り返すことができる。
本開示の局面に従って実現されるシステムは、スパース行列とベクトルとの反復乗算を伴う作業負荷をより効率的に実行することができる。たとえば、本開示の局面に係るシステムオンチップ(SoC:system-on-a-chip)を実現するデバイスは、少なくとも、最終的な積に寄与しない行列のゼロ要素の冗長な「ゼロ乗算」計算が省略されるので、従来のアプローチよりも少ない処理サイクルで、スパース行列にベクトルを乗算した積を生成することができる。他のアプローチでは、スパース比が高い大きな行列では性能損失が増大する可能性があるが、本明細書に記載されているように実現されるシステムは、少なくとも、入力行列のスパース比が高くなるにつれて省略される冗長計算の割合が高くなるので、これらの行列を用いてさらに効率的に積を計算することができる。
ニューラルネットワークの実行または訓練などの一部の作業負荷は、行列乗算の実行に大きく依存している。ハードウェアアクセラレータまたはその他のデバイスは、行列乗算などの特定の演算を効率的に実行することができるが、処理条件が制限されていることが多い。たとえば、デバイスは、行列のゼロ値と非ゼロ値との予め定義されたスパース比を要する場合があったり、処理する入力のサイズが厳しく制限されている場合がある。行列乗算は、多くのニューラルネットワーク作業負荷にとってユビキタスなタイプの計算であるが、処理条件が制限されているアクセラレータは、さまざまなスパース比の行列の行列乗算を伴う作業負荷など、アクセラレータがサポートできる作業負荷の種類の数が制限されている。
本開示の局面は、異なるサイズおよびスパース比の行列を柔軟に処理することができるスパース行列乗算のためのシステムを提供する。スパース行列乗算のために構成されたスパースシャードのシステムのスパースシャードは、複数の列にわたって所与の入力部分行列の非ゼロ値の位置を追跡し続けながら、入力行列のゼロ値を破棄することができる。スパースシャードは、予め定められた最大非ゼロ閾値と同数のレジスタおよび乗算回路のみで構成され、入力部分行列の非ゼロ値のみをメモリに格納することができる。その結果、スパースシャードは、フルサイズの行列を格納するのと比較して、より効率的に、より少ないリソースで、入力データを格納して処理することができる。
さらに、スパースシャードは、非ゼロ値と、それらに入力ベクトルの値を乗算した積とを、隣接する乗算回路および加算回路に沿ってそれぞれ配列するように構成され得る。スパースシャードは、これらの隣接する回路の出力を組み合わせることによって入力行列の列ごとにセグメント化合計を効率的に生成し、このセグメント化合計を再配列してシャード出力ベクトルを生成するように構成され得る。スパースシャードは、非ゼロ値の順序を保存して、正確に非ゼロ値を加算してこれに入力ベクトルの対応する値を乗算し、入力部分行列と入力ベクトルとの積としてシャード出力ベクトルを正確に生成することができる。スパースシャードは、入力行列の形状を変更するための前処理演算を必要とせずに、入力行列に対して乗算を実行することができる。
また、このシステムは、複数のスパースシャードによって入力ベクトルと乗算するためにスパース行列を前処理するように構成された1つ以上のプロセッサを含み得る。このシステムは、利用可能なスパースシャードの数に応じて異なる数の部分行列を生成すること
、および異なる最大非ゼロ閾値についての部分行列を生成することなど、スパースシャードの異なる構成をサポートすることができる。
本明細書に記載されている本開示の局面は、1つ以上のプロセッサと複数のスパースシャードとを含むシステムによって実現することができる。このシステムは、たとえば、コンピューティングプラットフォームのデータセンター内のサーバコンピューティングデバイスなどのコンピューティングデバイスにインストールされたチップとして実現することができる。
システムの例
図1は、スパースシャード101A~Pのアレイ101を含むシステム100の一例のブロック図である。アレイ101は、行列入力に対して行列乗算を実行するように構成されたスパース行列乗算システム100などのスパース行列乗算システムの少なくとも一部であってもよい。
図2を参照して本明細書により詳細に記載されているように、スパースシャードは、算術演算を実行するように、算術演算を実行する回路間に入力および出力を配列するように、ならびに/または他の回路間の入力および出力をゲート開閉するように構成された回路の集合体である。たとえば、スパースシャードは、スパース行列の矩形シャードまたは部分と、ベクトルの線形シャードまたは片との間の行列乗算を実行するように構成され得る。本明細書により詳細に記載されているように、スパースシャードは、予め定められた最大閾値までの任意の数の非ゼロ値を有する、スパース行列のさまざまな異なるシャードまたは部を受けることができる。
回路の集合体は、各ゲートの状態(すなわち、開もしくは閉)、計算されるオペランド、ならびに/または各回路への入力および各回路からの出力の配列を制御するように、本明細書に記載されているように構成され得る。この構成は、スパースシャードが入力として受けるように構成されているシャード入力行列の次元に、ならびに/またはシャード入力行列内の非ゼロ値の配列および数に、少なくとも部分的に依存し得る。各スパースシャードは、スパース行列にベクトルを乗算するように構成されているシステムに対する個々の回路コンポーネントであってもよい。
各スパースシャード101A~Pは、アレイ101内の2つ以上の他のスパースシャードと通信するように構成されている。スパースシャードは、そのすぐ隣のスパースシャード、たとえば、スパースシャード101A~Pの矩形配列によって定義される次元に沿った当該スパースシャードの直前または直後のスパースシャードと通信することができる。スパースシャード間の接続は、たとえば、スパースシャードをその近隣のスパースシャードに物理的に接続するバスまたは1つ以上の回路相互接続を介して実現することができる。各スパースシャード101A~Pは、行列の少なくとも一部およびベクトルの少なくとも一部を受け、入力行列とベクトルとを乗算した積を表す出力ベクトルを生成するように構成された1つ以上の回路として実現することができる。いくつかの例では、スパースシャード101A~Pはシストリックアレイとして編成されるが、さまざまな実現例において、アレイ101は一般にスパースシャード101A~Pの矩形配列に従って構成または配列される。
アレイ101は、回路基板または他の材料の上に複数のコンポーネントおよび集積回路を実現するシステムオンチップの少なくとも一部であってもよい。アレイ101は、コンピューティングデバイスの一部としてインストールされ、メモリ、プロセッサ、ネットワークコンポーネント、および/または周辺機器など、デバイスの他のコンポーネントとやり取りするように構成され得る。たとえば、アレイ101は、システム100の一部とし
て実現された1つ以上のメモリデバイスから、システム入力ベクトル105およびシステム入力行列110を受けることができる。システム100は、出力として、システム入力ベクトル105とシステム入力行列110とを乗算した積を表すシステム出力ベクトル115を生成することができる。
いくつかの例では、システム出力ベクトル115は、システム100を実現する他のデバイスまたはデバイスのコンポーネントへの入力として供給され得る。たとえば、システム出力ベクトル115が、モデルパラメータ値にニューラルネットワークについてのあるベクトル入力を乗算した積である場合は、システム出力ベクトル115は、出力ベクトル115の活性化関数を計算するように構成された1つ以上のプロセッサへの入力として供給され得る。
システム100は、前処理エンジン150からシステム入力ベクトル105およびシステム入力行列110を受けることができる。前処理エンジン150は、システム100を実現するデバイスであってもなくてもよい1つ以上のコンピューティングデバイス上に実現することができる。システム入力行列110は、一例として、ニューラルネットワークについてのモデルパラメータ値の少なくとも一部を表す値を含み得る。システム入力行列110は、多次元配列またはテンソルなどのより複雑なデータ構造の一部であってもよい。システム100は、3次元テンソルまたは行列などのより大きなデータ構造の少なくとも一部に対応する各行列を受け、この行列にシステム入力ベクトル105を乗算した対応する出力を生成するように構成され得る。システム入力ベクトル105は、たとえば、そのモデルパラメータ値がシステム入力行列110の値によって少なくとも部分的に表される訓練済みニューラルネットワークへの入力であってもよい。
前処理エンジン150は、システム入力行列110を処理して1つ以上の制御値111および行列パーティションデータ112を生成するように構成され得る。前処理エンジン150は、システム入力行列110を、スパース行列を格納するためのさまざまな異なるフォーマットで、たとえばその完全な行-列形式(すべての非ゼロ値およびゼロ値を有する)で受けることができる。他の例として、前処理エンジン150は、圧縮されたスパース列フォーマット、座標リストフォーマット、またはスパース行列を格納するためのさまざまな他のフォーマットのうちのいずれかに従って、システム入力行列110を受けることができる。
前処理エンジン150がシステム入力行列110をその完全な形で受けるいくつかの例では、前処理エンジン150は、システム入力行列110を、メモリに格納するのに適していると予め定められたフォーマットに変換するように構成されている。たとえば、前処理エンジン150は、行列からゼロ値を除去し、行列内の非ゼロ値の元の位置に対する非ゼロ値の位置を追跡するための制御値を生成してから、この予め定められたフォーマットに変換することができる。
制御値は、図2を参照して本明細書により詳細に記載されているように、システム入力ベクトル105の部分ベクトルを受けてこれにシステム入力行列110の部分行列を乗算するための各スパースシャードを構成するために使用される。パーティションデータ112は、どのようにシステム入力行列110を部分行列に区分すべきかを指定するデータである。各部分行列はそれぞれのスパースシャードにおいて入力として受けられ、パーティションデータ112は、アレイ101内のスパースシャードと同数の部分行列についての区分を指定する。
前処理エンジン150は、ベクトルパーティションデータ106を生成するように構成され得る。図5A~図5Cを参照してより詳細に記載されているように、同じ行または列
に沿ったスパースシャードは、同じ部分ベクトルを、それぞれの部分行列と乗算するための入力として受けることができる。
システム入力行列110およびシステム入力ベクトル105はシステム100の左側および右側に沿って供給されるものとして示されているが、システム100に入力データを供給するバスの正確な位置決めおよび方向付けは実現例によって異なり得る。たとえば、システム100と同じチップ上の他のコンポーネントの位置に基づいて、システム100に入力を供給してシステム100から出力を受けるためのバスまたは回路相互接続は、それらの他のコンポーネントの位置を考慮してさまざまに方向付けられ得るかまたは位置決めされ得る。
図2は、本開示の局面に係る、スパースシャード200の一例のブロック図である。たとえば、システム100のスパースシャード101A~Pの各々は、スパースシャード200を参照して本明細書に記載されているように実現することができる。
スパースシャード200は、シャード入力ベクトル205およびシャード入力行列210を受けるように構成されている。シャード入力ベクトル205は、1つ以上のベクトル値を含み、1×Rの最大次元を有し得る。シャード入力行列は、1つ以上のゼロ値および1つ以上の非ゼロ値を含み、R×Cの最大次元を有する。R(行)およびC(列)は、予め定められた、スパースシャードが入力として受けることができるベクトル/行列の最大入力サイズに対応する、次元閾値である。異なる実現例では、スパースシャードは異なる次元RおよびCについて構成され得る。RおよびCは互いに等しくても異なっていてもよく、たとえばスパースシャードが処理するように構成されている異なる作業負荷のデータの性質に応じて、異なる次元に異なるスパースシャードを実現することができる。スパースシャードのアレイのスパースシャードは、同じ最大次元閾値内の入力を受けるように構成され得る。
スパースシャード200は、スパースシャード200に関連付けられたメモリ内の予め定められたアドレス範囲内のシャード入力ベクトル205およびシャード入力行列210を受けるように構成され得る。たとえば、スパースシャード200は、結合されたメモリ内の第1のアドレス範囲からシャード入力ベクトル205を自動的に取り出し、メモリ内の同じまたは異なるアドレス範囲からシャード入力行列210を取り出すように構成されている。スパースシャード200を有するシステム100を実現するデバイスまたはデバイスのコンポーネントは、システムによって実現される1つ以上のスパースシャードの各々に対応するメモリ内の場所にシャード入力行列およびベクトルを送るように構成され得る。たとえば、前処理エンジン150は、システム入力行列および/またはシステム入力ベクトルを処理生成した後、個々のシャード入力行列およびシャード入力ベクトルを、スパースシャード200を含む各スパースシャードに対応するアドレス範囲に格納するように構成され得る。
最大次元閾値に加えて、スパースシャード200は、予め定められた最大非ゼロ閾値以下の非ゼロカウントを有するシャード入力行列を受けるように構成されている。次元閾値と同様に、最大非ゼロ閾値は、スパースシャード200およびその対応するアレイの実現例によって異なる値に設定され得る。たとえば、システムが処理するように構成される作業負荷のデータが一般に、スパース比が高い処理行列を含む場合、スパースシャードのシステムは、比較的高い最大非ゼロ閾値で構成され得る。本明細書に記載されているように、スパースシャードの乗算回路および加算回路の数は、その最大非ゼロ閾値に対応し、したがって、最大非ゼロ閾値が相対的に低いスパースシャードは、最大非ゼロ閾値が相対的に高いスパースシャードと比べて、少ない回路で構築することができる。
スパースシャード200は、クロスバー215および乗算回路220を含み得る。クロスバー215は、最大でR個のベクトル値を受け、その値をN個の乗算回路220に分散させるように構成され、Nはスパースシャード200についての最大非ゼロ閾値と等しい。乗算回路220A~C、Nが示されているが、異なる実現例ではスパースシャード200はより多いまたはより少ない乗算回路を含み得ると理解される。
クロスバー215は、入力を受けてその入力を1つ以上の宛先に渡すように構成された任意の1つまたは複数の回路として実現することができ、この宛先は乗算回路220などの他の回路であってもよい。乗算回路220は、2つのオペランド間のハードウェア乗算を実行するためのさまざまな異なる技術のうちのいずれかに従って実現することができる。乗算回路の第1のオペランドは、クロスバー215が受けたベクトル値であってもよい。乗算回路の第2のオペランドは、シャード入力行列210からの非ゼロ値であってもよい。各非ゼロ値は、それぞれの乗算回路220A~C、Nのそれぞれのレジスタ221A~C、Nにロードされる。最大非ゼロ閾値と同数の乗算回路を有することにより、非ゼロ閾値内のシャード入力行列ごとに利用可能な乗算回路が提供される。各乗算回路は、そのそれぞれのレジスタに格納された非ゼロ値に、クロスバー215が受けたベクトル値を乗算する。
また、スパースシャード200は加算回路225を含み得る。各加算回路225A~C、Nは、対応する乗算回路から入力を受けるように構成されている。各加算回路の間にセグメントマーカがある。加算回路225A~C、Nおよびセグメントマーカ226A~C、N-1が図2に示されているが、乗算回路220と同様に、加算回路およびセグメントマーカの数は実現例によって異なり得ると理解される。
加算回路は、2つのオペランドのハードウェア追加のための任意の技術を用いて実現することができる。加算回路からの第1のオペランドは、乗算回路から受けた積であってもよい。たとえば、乗算回路220Aは、ベクトル値と非ゼロ値とを乗算した積を加算回路225Aに渡す。セグメントマーカは、ゲート入力値に応じて、隣接する加算回路間の入力をゲート開閉するように構成された回路または他のハードウェアコンポーネントである。
制御値230は、シャード入力行列210およびシャード入力ベクトル205とともに入力として受けられ、クロスバー215、セグメントマーカ226、および/またはクロスバー235のうちの1つ以上を構成するために使用され得る。制御値230は、シャード入力行列210のそれぞれの非ゼロ値にそれぞれ対応する値のシーケンスであってもよい。制御値230は、1などの第1のタイプの値を含み得、これらの値は、シャード入力行列のそれぞれの列の第1の非ゼロ値である非ゼロ値に対応し得る。制御値230は、0などの第2のタイプの値を含み得、これらの値は、シャード入力行列の同じ列の1つ以上の他の非ゼロ値の後ろに位置する非ゼロ値に対応し得る。また、制御値230は、図3A~図3Cを参照して本明細書に記載されているように、シャード入力行列210およびシャード入力ベクトル205を処理するようにスパースシャードを構成するために本明細書に記載されているように使用され得る値の1つ以上のベクトルを含み得る。図4およびその対応する本明細書中の記載は、スパースシャードを用いる行列乗算の一例を示す。
いくつかの実現例では、クロスバー215、235は、スパースシャード200の異なる最大次元閾値を利用するように実現することができる。たとえば、次元RおよびCが等しいまたはほぼ等しい場合は、クロスバー235は、たとえばベネシュネットワークとして、正方形またはほぼ正方形の入力に対するクロスバー再配列のための任意の技術に従って実現することができる。
図3A~図3Cは、シャード入力行列300A、シャード入力行列内の非ゼロ値のベクトル300B、およびシャード入力行列700Aに対応する制御値のベクトル300Cの一例を示す。また、図3Cは追加の制御ベクトル305Cおよび310Cを示している。
図3Aは、スパースシャードのシャード入力行列300Aの一例を示す図である。分かりやすくするために、非ゼロ値は影付きセルとして示されている。分かりやすくするために、シャード入力行列300Aの列および行に沿って、かつベクトル300BおよびCに沿って、インデックスが与えられている。たとえば、行列300Aにおいて、行2、列4(2,4)の値は1である。
図3Bは、シャード入力行列300Aの一例の非ゼロ値のベクトル300Bを示す図である。ベクトル300Bの非ゼロ値は、シャード入力行列300Aの非ゼロ値を左から右に読み取ったときの出現順序に対応するが、正確な読み取り順は実現例ごとに異なり、たとえば右から左であってもよい。
図3Cは、シャード入力行列300Aの一例についての制御値のベクトル300C、305C、および310Cを示す図である。いくつかの例では、ベクトル300C、305C、および310Cは、予め定められた順序に従って同じベクトルの一部であってもよく、スパースシャードは、この予め定められた順序を用いて、本明細書に記載されているように、制御値のベクトルを受け、ベクトル300C、305C、および310Cに従ってスパースシャードの異なる成分を構成するように構成され得る。
ベクトル300Cは、スパースシャードのセグメントマーカを構成するために使用される制御値に対応する。ベクトル300Cにおける1の値の制御値(この例ではビット)は、行列300Aの新しい列の開始に対応する。インデックス0の値は、制御値のベクトル300Cの開始として、自動的に1に設定され得る。いくつかの実現例では、開始制御値は、処理スパースシャードによって省略されて定数と仮定され得る。ハードウェア実現例は、この値が1であることが既知であるという事実を利用することによって、その回路を簡素化することができる。ベクトル300Cにおけるインデックス1の値も、行列300Aの次の列の最初の非ゼロ値であるベクトル300Bにおけるインデックス1の値に対応するように、1に設定される。ベクトル300Cにおけるインデックス2の値は、次の列の最初の非ゼロ値ではないベクトル300Bにおける非ゼロ値に対応するので、0に設定される。
別の例として、ベクトル300Cにおけるインデックス3のビットは、対応する非ゼロ値(ベクトル300Bのインデックス3の値1)が行列300Aの次の列(具体的には列2)の最初の非ゼロ値であるので、1に設定される。ベクトル300Cにおけるシーケンスは、すべての列のすべての非ゼロ値が表されるまで、この説明したパターンに従う。
ベクトル305Cは、スパースシャードの入力クロスバーを構成するための制御値に対応する。ベクトル305Cは、シャード入力行列300Aの非ゼロ値ごとに、部分行列300A内の非ゼロ値の「y」座標を指定する。この例では、「y」次元はスパース行列300Aを垂直方向に上下するが、他の例では「y」次元はたとえば水平に定義されるなど、異なって定義され得る。たとえば、ベクトル305Cにおける要素ゼロの値「3」は、行列300Aの1列目の非ゼロ値「1」の「y」座標に対応する。別の例として、ベクトル305Cにおける要素6の値「4」は、行列300Aの4列目の一番下の値「1」に対応する。
ベクトル310Cは、スパースシャードの加算回路によって生成された合計が、シャード出力ベクトルを生成するために出力クロスバーによってどのように配列されるかを構成
するための制御値に対応する。
行列乗算の数学的定義により、スパース行列の位置(x,y)における各値に、位置yにおける入力ベクトルの値が乗算される。この乗算の結果は位置xにおける出力に加算される。入力クロスバーは、ベクトル305Cを用いて、同じ列の非ゼロ値をスパースシャード内の隣接する乗算回路に配列する。出力クロスバーは、ベクトル310Cを用いて、スパースシャードによってシャード入力行列とシャード入力ベクトルとを乗算した積を表すシャード出力ベクトルにおいて、計算した合計を正しい順序で配列する。
図2に戻って、クロスバー215は、制御値230を受け、シャード入力ベクトルにシャード入力行列が乗算されると、対応するシャード入力行列の列に一致する1つ以上の乗算回路への入力として受けられるようにシャード入力ベクトル205についての各値を配列するように構成され得る。
クロスバー235は、加算回路から1つ以上の合計を受け、受けた合計を再配列して、入力シャード行列と入力シャードベクトルとの乗算に対応する正しい出力シャードベクトルを得るように構成され得る。クロスバー215と同様に、クロスバー235を1つ以上の回路として実現するためのさまざまな異なる技術のうちのいずれかを適用することができる。
本明細書に記載されているように、セグメントマーカは、セグメントマーカのゲート入力値に応じて、隣接する加算回路間の入力をゲート開閉するように構成されている。たとえば、セグメントマーカが1の値の制御値を受けた場合は、セグメントマーカは、セグメントマーカに隣接する第1の加算回路からの出力が、セグメントマーカに隣接する第2の加算回路への入力として渡されることを防止することができる。セグメントマーカが0の値の制御値を受けた場合は、セグメントマーカは、第1の加算回路からの出力を第2の加算回路に(または、実現例によってはその逆に)渡す。このセグメントマーカの構成は、異なる列の非ゼロ値の合計とは別に、同じ列の非ゼロ値に対応する合計のみを加算することに相当する。図2を参照して本明細書に記載されているように、この1つ以上の合計は、クロスバー235に渡され、シャード出力ベクトル240を生成するように再配列され得る。
クロスバー235は、入力を破棄するか受けるかを判断し、その入力をシャード出力ベクトル240内のその正しい位置に一致するように再配列するように構成され得る。図2に示されるように、各加算回路はクロスバー235に出力を渡すことができる(クロスバー235への矢印で示す)。ある加算回路の後ろの次のセグメントマーカがゲート開閉されていない場合は、隣接する加算回路間のランニングサムが終了していないので、クロスバー235はその加算回路への出力を破棄することができる。次のセグメントマーカがゲート開閉されている場合(または、最後の加算回路225Nの場合はセグメントマーカがない場合)は、その列についてのランニングサムは終了しており、クロスバー235は、出力シャードベクトル240の一部になるべき入力としてランニングサムを受ける。どの合計を無視し、どの合計をシャード出力ベクトル240の一部として含めるかを把握することによって、クロスバー235は、最大次元Cまで、列和を正確に追跡してさまざまな長さの出力ベクトルを生成することができる。
図4は、シャード入力行列410と、シャード入力ベクトル405と、シャード入力行列410についての制御値430とを受けるスパースシャード400の計算の一例を示す。
シャード入力ベクトル405およびシャード入力行列410の値の一例について考える
[1 3 2](ベクトル405)
[0 3 0 2 0 4 1 0 0](行列410)
この例における行列410についての対応する制御ベクトル430~432は、
[1 0 1 1](制御ベクトル430)
[1 2 0 1](制御ベクトル431)
[0 0 1 2](制御ベクトル432)
である。
説明し易くするために、乗算回路40A~Dを乗算器A~Dと略し、セグメントマーカ43A~CをセグメントマーカA~Cと略し、加算回路42A~Dを加算器A~Dと略することにする。
行列410の非ゼロ値に基づいて、乗算器Aには値2がロードされ、乗算器Bには値1がロードされ、乗算器Cには値3がロードされ、乗算器Dには値4がロードされる。なお、この例では、スパースシャードは4つの乗算器A~Dおよび4つの加算器A~Dのみを含む。
クロスバー415は、ベクトル値1、3、および2を有するベクトル405を受ける。シャード入力ベクトル405からシャード出力ベクトル440へのデータの経路を示すために、破線45A、実線45B、および点線45Cが示されている。クロスバー415は制御ベクトル431を受け、制御ベクトル431の各値はシャード入力行列410のそれぞれの非ゼロ値の「y」座標に対応する。制御ベクトル431の値は、1、2、0、および1を含む。なお、スパース入力行列410は次元3×3を有するので、「y」座標の値は0から2の範囲である。制御ベクトル431は、乗算器A~Dの各々について、シャード入力ベクトル405の値のうちのどの値をどの乗算器に送るべきかを指定する。
たとえば、制御ベクトル431の最初の値は1であり、シャード入力行列410の最初の非ゼロ値の「y」座標に対応する。「y」座標1は(ゼロの後の)2番目の座標であるので、クロスバー415は、シャード入力ベクトル405の2番目の値を第1の乗算器(ここでは乗算器A)に導く。制御ベクトル431の2番目の値は2であり、シャード入力行列410の2番目の非ゼロ値の「y」座標に対応する。クロスバー405は、次に、シャード入力ベクトル405の3番目の値を乗算器Bに導くように構成され得る。別の例として、制御ベクトル431の3番目の値は0であり、0の「y」座標を有する次の非ゼロ値に対応する。クロスバー415は、シャード入力ベクトル405の最初の値を乗算器Cに導く。
行列410の1列目について、クロスバー415は値3を乗算器Aに導き、値2を乗算器Bに導く。行列410の2列目について、クロスバー415は値1を乗算器Cに導く。行列410の最後の3列目について、クロスバー415は値3を乗算器Dに導く。乗算器A~Dの積は、乗算器Aは6(3×2)、乗算器Bは2(2×1)、乗算器Cは3(1×3)、乗算器Dは12(3×4)である。
次に、乗算器A~Dが計算した積を加算器A~Dが受ける。加算器Aは、乗算器Aからの積、つまり6を受ける。最初の制御値(1)は破棄される。加算器Aの前には加算器がないので、加算器AはセグメントマーカAに合計を渡す。制御値430の2番目の値はゼロであるので、セグメントマーカAはゲート開閉されていない。加算器Bは、加算器Aから現在の合計(6)を受け、それを乗算器Bの積(2)に加算する。セグメントマーカBはゲート開閉されているので、加算器Bの出力(8)はクロスバー420に渡される(破
線45Aで示す)。セグメントマーカBは加算器Bからの出力をゲート開閉するので、加算器Cは乗算器Cの積(3)にゼロを加算する。セグメントマーカCはゲート開閉されているので、加算器Cの出力(3)はクロスバー420に渡される(点線45Cで示す)。最後に、乗算器Dは、乗算器Dの積(12)を受け、これはスパースシャード内の最後の加算器であるので、その出力(12)をクロスバー420に自動的に渡す(実線45Bで示す)。
クロスバー420は、受けた合計8、3、および12を、出力シャードベクトル440を出力するための正しい順序に従って再配列する。クロスバー420は、値0、0、2、および1を有する制御ベクトル432を受ける。図3A~図3Cを参照して本明細書に記載されているように、出力クロスバー420についての制御ベクトルの値は、非ゼロ値の「x」座標位置に対応する。「y」座標と同様に、この値はこの例では0から2の範囲である。制御ベクトル432の最初の2つの値は0である。したがって、クロスバー420は、受けた第1の合計を出力シャードベクトル440の第1の要素に導く。クロスバー432の0の次の値は2である。クロスバー432は、加算器A~Dから受けた第2の合計を出力ベクトル440の第2の要素に導き(線45Cで示す)、第3の合計を要素に導く(線45Bで示す)ように構成されている。いくつかの例では、クロスバー420は、たとえば最初の2つがゼロであるベクトル432に示されるような、ベクトル432の連続する重複制御値をスキップするように構成されている。いくつかの例では、連続する重複制御値をスキップするのではなく、クロスバー420は、受けた入力合計に対して包含的OR演算を実行し、この包含的OR演算の結果を、連続する重複制御値に対応する出力ベクトル440の位置に出力するように構成されている。
たとえば、ライン46は、出力クロスバー420への加算器Aの潜在的な入力ソースを示す。セグメントマーカ43Aはゼロの値を有するので、クロスバー420への加算器Aの出力は抑制され、たとえばマスクされるかまたはゼロに設定される。代わりに、加算器Aの出力はセグメントマーカ43Aを通って加算器Bに渡される。いくつかの例では、出力クロスバー420が制御値432を受けると、出力クロスバー420は、(ライン46を通して)受けた第1の合計(ゼロの値を有する)と、(ライン45Aを通して)受けた第2の合計(加算器Bからの8の値を有する)とに対して包含的OR演算を実行する。クロスバー420は、非ゼロオペランドを出力するために包含的OR演算を実行するように構成され得る。クロスバー420は、受けた合計に対して包含的OR演算(たとえば、0
OR 8)を実行した後に8を出力し、その結果を出力ベクトル440内の最初の位置に渡す。いくつかの例では、出力クロスバー420は加算器A~Dのうちの少なくともいくつかから個々の出力を受けてそれらを合計することができる。
加算回路は、セグメントを定義する連続した数値範囲を追加するためのさまざまな異なる回路構成のうちのいずれかに従って実現することができ、各セグメントは、対応するスパースシャードによって処理されるシャード入力行列のそれぞれの列内の値に対応する。たとえば、スパースシャード200または400の加算回路は、各列の非ゼロ値にシャード入力ベクトルのそれぞれの値を乗算することによって積の順次セグメント化合計を実行するための、(たとえば、スパースシャード200または400によって示されるような)1つ以上の順次セグメント化合計回路として実現することができる。いくつかの実現例では、加算回路は、並列セグメント化合計回路として合計ツリーを実行するように構成され得る。個々の加算回路は、本明細書に記載されているように、対応する入力を並列に加算し、その合計を、任意の介在するセグメントマーカのゲート値に従って、出力クロスバーおよび/または隣接する加算回路に渡すように構成され得る。
並列セグメント化合計回路は、特に合計している項の数が多い場合に、順次セグメント化合計回路と比べて回路の待ち時間を短縮することができる。セグメントマーカをゲート
開閉するための制御値によって、並列セグメント化合計が可能になる。なぜなら、少なくとも、合計する値の範囲を、異なるセグメント内の値に対応する加算回路間の入力をゲート開閉するとともに同じセグメント内の追加される値に対応する加算回路間の入力を可能にする、セグメントマーカのゲート値に従って追跡することができるからである。
いくつかの実現例では、出力クロスバー420は、各加算器によって計算された個々の合計を受け、その合計を加算し、制御ベクトル432を用いて、その合計を出力シャードベクトル440の対応する要素に導くように構成されている。2つ以上の合計が同じ要素に導かれる場合、たとえば、制御ベクトル432がベクトル432の最初のゼロのように同じ値の重複を含む場合は、出力クロスバーは、図4の線45Aで示すように1つの合計を受けるのではなく、出力シャードベクトル440内の同じ要素に導かれるように、受けた各合計を加算するように構成されている。
方法の例
図5は、本開示の局面に係る、スパースシャード上のスパース行列の部分行列にシステム入力ベクトルを乗算するためのプロセス500の一例のフロー図である。説明し易くするために、スパース行列の部分行列をシャード入力行列と呼ぶ。図9は、本明細書において、入力スパース行列を複数の部分行列に区分するためのプロセスの一例を示す。たとえば、図2のスパースシャード200などのスパースシャードがプロセス500を実行する。
ブロック510に従って、スパースシャードはシャード入力行列を受ける。シャード入力行列は、予め定められた次元閾値内にあり、予め定められた最大非ゼロ閾値以下の非ゼロ値カウントを有する。
ブロック520に従って、スパースシャードは、複数のベクトル値を含むシャード入力ベクトルを受ける。シャード入力ベクトルは、システム入力ベクトルの部分ベクトルであり、これは、図1および図9を参照して本明細書に記載されているように、前処理エンジンによってシステムに対する前処理入力の一部として生成することができる。
ブロック530に従って、スパースシャードは、それぞれのベクトル値にそれぞれの非ゼロ値を乗算した1つ以上の積を生成する。図2および4を参照して本明細書に記載されているように、スパースシャードは、シャード入力行列内の非ゼロ値の位置に対応する制御値で構成され得る。この制御値に基づいて、スパースシャードは、部分ベクトルの受信ベクトル値を、入力シャード行列の同じ列に沿った非ゼロ値が格納されている、対応する隣接する乗算回路に導くように構成され得る。本明細書に記載の図8Aは、シャード入力行列についての制御値を用いてスパースシャードを構成するためのプロセスの一例を示す。
ブロック540に従って、スパースシャードは、この1つ以上の積の1つ以上の合計を生成する。図2を参照して本明細書に記載されているように、スパースシャードは、対応する乗算回路から入力を受けるように構成された複数の加算回路を含む。加算回路はさらに、セグメントマーカに隣接する回路をゲート開閉するように設定された当該マーカに達するまで、隣接する加算回路に沿ってこれらの入力を加算するように構成されている。最大でセグメントマーカまでの加算回路入力の合計は、たとえば図2のクロスバー235などのクロスバーに渡され得る。
スパースシャードは、この1つ以上の合計から、シャード入力ベクトルにシャード入力行列を乗算した積であるシャード出力ベクトルを生成する。スパースシャードは、乗算回路によって生成された積のオペランドが第1のクロスバーによってどのように順序付けら
れたかに応じて、たとえば第2のクロスバーを通して、受けた合計を再配列することができる。
図6は、複数のスパースシャードからのシステム入力ベクトルとシステム入力行列との積を表すシステム出力ベクトルを生成するプロセス600の一例のフロー図である。図1のスパース行列乗算システム100などのスパースシャードのシステムが、プロセス600を実行することができる。
ブロック610に従って、スパースシャードのアレイの列次元に沿ったスパースシャードのグループごとに、グループ内の各スパースシャードのシャード出力ベクトルを加算してそのグループの列出力ベクトルを生成する。後述する図7Cは、それぞれのシャード入力行列に従うスパースシャードのグループ化の一例を示す。スパースシャードのグループが沿って形成される次元は、実現例によって異なり得る。たとえば、システム入力行列およびシステム入力ベクトルがスパースシャードのアレイに供給される方向に応じて、グループは、列とは対照的に、アレイの行に沿うことができる。
ブロック620に従って、システムは、各列出力ベクトルを連結してシステム出力ベクトルを生成する。システム出力ベクトルは、システム入力行列にシステム入力ベクトルを乗算した積である。いくつかの実現例では、図9を参照して本明細書により詳細に記載されているように、システムは、行列の列が置換されるシステム入力行列を受けて、たとえば、非ゼロ値の出現をスパースシャードに割り当てられた部分行列により均等に分散させることができる。それらの実現例では、システムは、連結されたシステム出力ベクトルの要素を再配列して元の順列を逆にするように構成され得る。システムは、システム入力行列、制御値、およびシステム入力行列の区分を定義するデータを受ける一部として、並び替えを定義するデータを受けることができる。
図7A~図7Cおよび対応する記載は、スパース行列700の一例とベクトル750との間の乗算の一例を示す。説明のために、この乗算は、4×4アレイの16個のスパースシャードを有するシステムに対して実行されるものとして記載されている。
図7Aは、システム入力行列700の一例およびシステム入力ベクトル750を示す図である。この図では、システム入力行列700は整数値で示されているが、要素はたとえば浮動小数点値などの他の値であってもよいと理解される。また、図7A~図7Cでは、システム入力行列700を含むさまざまな行列の非ゼロ値の要素は影付きセルとして示されている。
図7Bは、システム入力行列700およびシステム入力ベクトル750の区分を示す図である。この例では、システム入力行列700は、4×4アレイのスパースシャードごとに1つの部分行列があるように、16個の部分行列700A~Pに区分されている。システム入力ベクトル750は、4×4アレイのスパースシャードの列ごとに1つの部分ベクトルがあるように、4つの部分ベクトル750A~Dに区分されている。
部分ベクトルおよび部分行列を16個のスパースシャード(スパースシャードA~Pと呼ぶ)にマッピングする一例は、以下の表1に示す通りである。
Figure 2024028901000002
図7Cは、区分されたシステム入力行列700、およびシステム入力行列とシステム入力ベクトル750とを乗算した積を表すシステム出力ベクトル770を示す図である。図7Cは、列705A~Dに沿ってグループ化されている、区分された行列を示している。図6を参照して本明細書に記載されているように、システムは、スパースシャードのアレイの列ごとにシャード出力ベクトルを加算し、列出力ベクトルを生成することができる。図7Cにおいて、列出力ベクトル710A~Dは列705A~Dに対応する。システムは、列出力ベクトル710A~Dを連結して、システム入力行列にシステム入力ベクトルを乗算した積としてシステム出力ベクトルを生成することができる。
図8A~図8Bは、本開示の局面に係る、入力行列の1つ以上の制御値を用いてスパースシャードを構成し、行列乗算を実行するためのプロセス600A~Bの一例のフロー図である。
図8Aは、本開示の局面に係る、シャード入力行列の1つ以上の制御値を用いてスパー
スシャードを構成するためのプロセス800Aの一例のフロー図である。たとえば図2のスパースシャード200などのスパースシャードが、プロセス800Aを実行することができる。
ブロック810に従って、スパースシャードは、シャード入力行列の各列に沿った非ゼロ値の位置を指定する1つ以上の制御値を受ける。
ブロック820に従って、スパースシャードは、シャード入力行列の非ゼロ値を乗算回路のレジスタにロードする。図2を参照して本明細書に記載されているように、スパースシャードは、システムについて予め定められた最大非ゼロ閾値と等しい数の乗算回路を実現することができる。スパースシャードは、左から右などの予め決められた読み取り方向に沿ってシャード入力行列を読み取ったときの出現順序で非ゼロ値をロードすることができる。
ブロック830に従って、スパースシャードは、この1つ以上の制御値をスパースシャードのクロスバーにロードする。図2のクロスバー815などの第1のクロスバーは、制御値を受け、行列-ベクトル乗算の一部としてベクトル値が乗算される、対応するシャード入力行列の列に一致する1つ以上の乗算回路への入力として受けられるようにシャード入力ベクトルの各値を配列するように構成され得る。図2のクロスバー235などの第2のクロスバーは、加算回路から1つ以上の合計を受け、受けた合計を再配列して、入力シャード行列と入力シャードベクトルとの乗算に対応する正しい出力シャードベクトルを得るように構成され得る。
ブロック840に従って、スパースシャードは、この1つ以上の制御値を、この制御値の値に基づいて加算回路への入力をゲート開閉するように構成された1つ以上のセグメントマーカにロードする。
図8Bは、図8Aのプロセス800Aに従って構成されたスパースシャードを用いて行列-ベクトル乗算を実行するためのプロセスの一例のフロー図である。
ブロック850に従って、スパースシャードは、入力シャード行列の非ゼロ値を受けてロードする。
ブロック860に従って、スパースシャードは、シャード入力ベクトルのベクトル値を受け、これらのベクトル値をシャード入力行列の同じ列に沿った非ゼロ値とともに乗算回路に送る。
ブロック870に従って、スパースシャードは、セグメントマーカによってゲート開閉されていない隣接する加算回路から1つ以上のセグメント化合計を生成する。図2および図8Aを参照して本明細書に記載されているように、スパースシャードは、セグメントマーカによってゲート開閉されていない加算回路間の合計を集約し、シャード入力行列の異なる列からの計算を表す加算回路をゲート開閉するように制御値を用いてセグメントマーカを構成することができる。スパースシャードがアクティブゲートビットを有するセグメントマーカに達すると、スパースシャードは、セグメント化合計を再配列してシャード出力ベクトルを生成するように構成された第2のクロスバーにセグメント化合計を渡す。
ブロック880に従って、スパースシャードは、この1つ以上のセグメント化合計からシャード出力ベクトルを生成する。図2および図8Aを参照して本明細書に記載されているように、クロスバーは、スパースシャードの加算回路から1つ以上のセグメント化合計を受けるように、制御値を用いて構成され得る。クロスバーはさらに、セグメント化合計
を再配列して、シャード入力行列にシャード入力ベクトルを乗算した積を表す正しいシャード出力ベクトルを生成するように構成され得る。
図9は、本開示の局面に係る、システム入力行列から部分行列を生成するためのプロセス900の一例のフロー図である。1つ以上の場所にある1つ以上のプロセッサが、プロセス900を実行することができる。たとえば、図1の前処理エンジン150などの前処理エンジンが、プロセス900を実行することができる。
ブロック910に従って、前処理エンジンはシステム入力行列を受ける。システム入力行列は、たとえば、図7Aのシステム入力行列700で示されるような行列であってもよい。
ブロック920に従って、前処理エンジンは、システム入力行列を複数の候補部分行列に区分する。区分の一部として、前処理エンジンは、予め定められた次元閾値を指定するパラメータと、部分行列を受けるシステムによって実現されるスパースシャードの数を示す1つ以上のパラメータ値とを受けることができる。たとえば、前処理エンジンは、8行×8列の次元閾値内で(4×4アレイのスパースシャードについて)16個の候補部分行列を生成するように構成され得る。いくつかの例では、前処理エンジンは、たとえば、異なる次元閾値および/またはスパースシャードの構成を有する異なるシステムにわたる前処理入力のために、更新されたパラメータ値を受けることができる。
いくつかの実現例では、前処理エンジンがシステム入力行列を区分する前に、前処理エンジンはシステム入力行列の列を置換して、非ゼロ値を候補部分行列に均等に分散させる。たとえば、非ゼロ値が、予め定められた許容差を超えて入力行列の一方側よりも他方側に頻繁に出現する場合、前処理エンジンは、非ゼロ値の出現がより広がることにより、区分後にスパースシャードにより均等に分散するように、入力行列の列の順序を変更するように構成され得る。
前処理エンジンがこの順序付けを実行する場合、前処理エンジンは、順序付けを表すデータを、追加入力として、たとえばスパースシャードのアレイを有するシステムに送られるパーティションデータの一部として、システムに渡す。システムは、出力ベクトルを、システム入力行列の列が置換される前のシステム入力行列とシステム入力ベクトルとを乗算した出力と一致させるために、順序付けに従ってシステム出力ベクトルの要素を並べ替えるように構成され得る。
システム入力行列の列を置換すると、システム入力行列がスパースシャードのシステムによって処理される全体速度を向上させることができる。たとえば、列を置換することによって、特に、いくつかの例では、一部のスパースシャードがゼロ値のみを有するシャード入力行列を受け得、他のシャードが非ゼロ値のみを有するシャード入力行列、または非ゼロ閾値までの数の非ゼロ値を有するシャード入力行列を受け得る場合に、各スパースシャードをより効率的に使用することができる。
ブロック930に従って、前処理エンジンは、予め定められた非ゼロ閾値よりも大きい非ゼロ値カウントを有する候補部分行列があるか否かを判定する。予め定められた非ゼロ閾値よりも大きい非ゼロ値カウントを有する候補部分行列があると前処理エンジンが判定した場合は、ブロック940に従って、前処理エンジンは、候補部分行列と同じ行または列に沿って部分行列を再区分する。
図7Bに示されるように、部分行列は、入力行列におけるそれらの値の位置に基づいて、列および行に沿って編成され得る。たとえば、部分行列700Jが非ゼロ閾値よりも高
い非ゼロ値カウントを含むと前処理エンジンが判定した場合は、前処理エンジンは、部分行列700Jの行(部分行列700I、700K、および700Lを含む)に沿って、ならびに/または部分行列700Jの列(部分行列700B、700F、および700Nを含む)に沿って、部分行列を再区分することができる。再区分を実行する際、前処理エンジンは、予め定められた次元閾値を使用し、候補部分行列の数が変わらないように再区分を実行する。たとえば、前処理は、候補部分行列を分割し、決定された候補部分行列の行/列に沿って部分行列の行/列を再分散させることができる。
ブロック930および940に従って、部分行列を再区分した後、前処理エンジンは、非ゼロ閾値よりも大きい非ゼロ値カウントを有する候補部分行列があるか否かを再び判定する。前処理エンジンは、最大非ゼロ閾値を超える非ゼロ値カウントを有する候補部分行列がないと判定するまで、ブロック930および940に従って判定および再区分を繰り返し、ブロック950に進むことができる。
ブロック950に示されるように、前処理エンジンはシステム入力ベクトルを区分する。システム入力ベクトルの各部分ベクトルは、スパースシャードのアレイを供給するバスがどのように配列されるかに応じて、スパースシャードの行または列の各々に入力される。前処理エンジンは、ベクトル次元に、受けたスパースシャードの行列を乗算できるように、たとえば数学的に有効な行列乗算のための有効な次元を乗算できるように、ベクトルを区分する。
たとえば図7B~図7C、および表1に示されるように、ベクトル750は部分ベクトル750A~Dに区分され、これらの各々が入力として1つ以上のスパースシャードに渡される。また、図7Bでは、各部分ベクトル750A~Dの各次元は、対応する部分行列700A~Pと乗算するための正しい次元を有している。たとえば、部分ベクトル750Bは1×2(行×列)であり、部分行列700E~Hの各々は2行を有しているため、部分ベクトル750Bと部分行列700E~Hとの間の有効な行列乗算が可能である。
ブロック960に示されるように、前処理エンジンは、候補部分行列ごとに制御値を生成する。図3A~図3Cを参照して記載されているように、前処理エンジンは、行列の列ごとに開始非ゼロ値を示す制御値のベクトルを生成することができる。前処理エンジンは、候補部分行列ごとにこの生成を繰り返して、部分行列ごとに対応する制御値を生成する。
ブロック970に従って、前処理エンジンは制御値および候補部分行列を出力する。前処理エンジンは、制御値と、システム入力行列の区分を指定するデータとを、たとえば図1に示されるようなシステム100に出力することができる。
コンピューティング環境の例
図10は、本開示の局面に係る、スパース行列乗算システム100および前処理エンジン150を実現するコンピューティング環境の一例のブロック図である。前処理エンジン150は、サーバコンピューティングデバイス1015などにおいて、1つ以上の場所に1つ以上のプロセッサを有する1つ以上のデバイス上に実現することができる。ユーザコンピューティングデバイス1012およびサーバコンピューティングデバイス1015は、ネットワーク1060を介して1つ以上のストレージデバイス1030に通信可能に結合され得る。ストレージデバイス(複数可)1030は、揮発性メモリと不揮発性メモリとの組み合わせであってもよく、コンピューティングデバイス1012、1015と同じまたは異なる物理的位置にあってもよい。たとえば、ストレージデバイス(複数可)1030は、ハードドライブ、ソリッドステートドライブ、テープドライブ、光ストレージ、メモリカード、ROM、RAM、DVD、CD-ROM、書き込み可能メモリ、および読
み取り専用メモリなどの、情報を格納することができる任意の種類の非一時的なコンピュータ読取可能媒体を含み得る。
サーバコンピューティングデバイス1015は、1つ以上のプロセッサ1013およびメモリ1014を含み得る。メモリ1014は、プロセッサ(複数可)1013によって実行され得る命令1021を含む、プロセッサ(複数可)1013がアクセス可能な情報を格納し得る。また、メモリ1014は、プロセッサ(複数可)1013が取り出す、操作する、または格納することができるデータ1023を含み得る。メモリ1014は、揮発性メモリおよび不揮発性メモリなどの、プロセッサ(複数可)1013がアクセス可能な情報を格納することができる非一時的なコンピュータ読取可能媒体の一種であってもよい。プロセッサ(複数可)1013は、1つ以上の中央処理装置(CPU)、グラフィックス処理ユニット(GPU)、フィールドプログラマブルゲートアレイ(FGPA)、および/またはテンソル処理ユニット(TPU)などの特定用途向け集積回路(ASIC)を含み得る。
サーバコンピューティングデバイス1015は、スパース行列乗算システム100を、たとえばシステムオンチップとして、ハードウェア内に実現することができる。システム100は、サーバコンピューティングデバイス1015にスロットインされたまたはインストールされた物理的チップの一部として実現することができる。システム100は、サーバコンピューティングデバイス1015の他のコンポーネントと通信するように構成されている。
命令1021は1つ以上の命令を含み得、この命令は、プロセッサ(複数可)1013によって実行されると、この1つ以上のプロセッサに、この命令によって定義された動作を実行させる。命令1021は、プロセッサ(複数可)1013によって直接処理するためのオブジェクトコード形式で、または、解釈可能なスクリプトもしくはオンデマンドで解釈されるか事前にコンパイルされる独立したソースコードモジュールの集合体を含む他の形式で、格納され得る。命令1021は、本開示の局面と一致するスパースシャード400を実現するための命令を含み得る。プリプロセッサエンジン105は、プロセッサ(複数可)1013を用いて、および/またはサーバコンピューティングデバイス1015から離れて位置する他のプロセッサを用いて実行され得る。
データ1023は、命令1021に従ってプロセッサ(複数可)1013によって取り出され、格納され、または修正され得る。データ1023は、コンピュータレジスタ内に、複数の異なるフィールドおよびレコードを有するテーブルとしてリレーショナルもしくは非リレーショナルデータベース内に、またはJSON、YAML、proto、もしくはXML文書として、格納され得る。また、データ1023は、2進値、ASCIIまたはユニコードなどであるがこれらに限定されないコンピュータ読取可能形式にフォーマットされ得る。さらに、データ1023は、数字、説明文、プロプライエタリコード、ポインタ、他のネットワーク場所を含む他のメモリに格納されたデータの参照などの、関連情報を識別するのに十分な情報、または関連データを計算する機能によって使用される情報を含み得る。
ユーザコンピューティングデバイス1012も、1つ以上のプロセッサ1016と、メモリ1017と、命令1018と、データ1019とを有して、サーバコンピューティングデバイス1015と同様に構成され得る。また、ユーザコンピューティングデバイス1012は、ユーザ出力1026およびユーザ入力1024を含み得る。ユーザ入力1024は、キーボード、マウス、機械的アクチュエータ、ソフトアクチュエータ、タッチスクリーン、マイクロフォン、およびセンサなどの、ユーザから入力を受けるための任意の適切なメカニズムまたは技術を含み得る。
サーバコンピューティングデバイス1015は、ユーザコンピューティングデバイス1012にデータを送信するように構成され得、ユーザコンピューティングデバイス1012は、ユーザ出力1026の一部として実現されたディスプレイに受信データの少なくとも一部を表示するように構成され得る。また、ユーザ出力1026を用いて、ユーザコンピューティングデバイス1012とサーバコンピューティングデバイス1015との間のインターフェースを表示することができる。ユーザ出力1026は、これに代えてまたはこれに加えて、1つ以上のスピーカ、トランスデューサまたは他の音声出力、ユーザコンピューティングデバイス1012のプラットフォームユーザに非視覚情報および非可聴情報を提供するハプティックインターフェースまたは他の触覚フィードバックを含み得る。
図10は、プロセッサ1013、1016およびメモリ1014、1017がコンピューティングデバイス1015、1012の内部にあると示しているが、プロセッサ1013、1016およびメモリ1014、1017を含む本明細書に記載されているコンポーネントは、異なる物理的位置で動作することができ、かつ同じコンピューティングデバイスの内部にない、複数のプロセッサおよびメモリを含み得る。たとえば、命令1021、1018およびデータ1023、1019の一部は、取り外し可能なSDカードに格納され、他の部分は読み取り専用のコンピュータチップ内に格納されてもよい。命令およびデータの一部またはすべてが、プロセッサ1013、1016から物理的に離れているがプロセッサ1013、1016がアクセス可能な場所に格納されてもよい。同様に、プロセッサ1013、1016は、同時および/または順次動作を実行することができるプロセッサの集合体を含み得る。コンピューティングデバイス1015、1012の各々は、タイミング情報を提供する1つ以上の内部クロックを含み得て、このタイミング情報は、コンピューティングデバイス1015、1012によって実行される動作およびプログラムの時間測定に使用され得る。
サーバコンピューティングデバイス1015は、データの処理要求をユーザコンピューティングデバイス1012から受けるように構成され得る。たとえば、環境1000は、プラットフォームサービスを公開するさまざまなユーザインターフェースおよび/またはAPIを介して、さまざまなサービスをユーザに提供するように構成されたコンピューティングプラットフォームの一部であってもよい。サービスを実行する一部として、サーバコンピューティングデバイス1015は、システム100を用いて受信データを処理し得る。たとえば、サービスが機械学習モデルを訓練している場合は、サーバコンピューティングデバイス1015は、システム100を用いて、機械学習モデルを訓練する一部として乗算演算を実行するように構成され得る。
デバイス1012、1015は、ネットワーク1060を介した直接および間接通信が可能であってもよい。デバイス1015、1012は、情報を送受信するための開始接続を受け入れ得るリスニングソケットをセットアップしてもよい。ネットワーク1060自体が、インターネット、ワールドワイドウェブ、イントラネット、仮想プライベートネットワーク、ワイドエリアネットワーク、ローカルネットワーク、および1つ以上の企業に独自の通信プロトコルを用いるプライベートネットワークなど、さまざまな構成およびプロトコルを含み得る。ネットワーク1060はさまざまな近距離および長距離接続をサポートし得る。この近距離および長距離接続は、2.402GHz~2.480GHz(一般にブルートゥース(登録商標)規格に関連する)、2.4GHzおよび5GHz(一般にWi-Fi(登録商標)通信プロトコルに関連する)などの異なる帯域幅にわたって行われてもよく、または無線ブロードバンド通信用のLTE(登録商標)規格などのさまざまな通信規格で行われてもよい。また、ネットワーク1060は、これに加えてまたはこれに代えて、さまざまなタイプのイーサネット(登録商標)接続を介するなどして、デバイス1012、1015間の有線接続をサポートし得る。
図10には1つのサーバコンピューティングデバイス1015およびユーザコンピューティングデバイス1012が示されているが、本開示の局面は、順次もしくは並列処理のためのパラダイムにおいて、または複数の装置の分散ネットワークを介してなど、さまざまな異なる構成および量のコンピューティングデバイスに従って実現することができると理解される。いくつかの実現例では、本開示の局面は、1つのデバイス、およびその任意の組み合わせに対して実行され得る。さらに、前処理エンジンおよびスパース行列乗算システム100は同じサーバコンピューティングデバイス1015上に実現されるものとして示されているが、いくつかの実現例では、前処理エンジン150は、サーバコンピューティングデバイス1015とは別の、1つ以上のサーバコンピューティングデバイスおよび/またはユーザコンピューティングデバイス1012上に実現される。
本開示の局面は、デジタル回路、コンピュータ読取可能記憶媒体において、1つ以上のコンピュータプログラム、または上述のうちの1つ以上の組み合わせとして実現することができる。コンピュータ読取可能記憶媒体は、たとえば、コンピューティングデバイスによって実行可能であり有形ストレージデバイスに格納される1つ以上の命令として、非一時的であってもよい。
本明細書において、「~ように構成される」という表現は、コンピュータシステム、ハードウェア、またはコンピュータプログラム、エンジン、もしくはモジュールの一部に関連する異なる文脈で使用されている。システムが1つ以上の動作を実行するように構成されると言う場合、これは、動作時に、システムにこの1つ以上の動作を実行させる適切なソフトウェア、ファームウェア、および/またはハードウェアが、システムにインストールされていることを意味する。あるハードウェアが1つ以上の動作を実行するように構成されると言う場合、これは、動作時に、入力を受け、この入力に従ってこの1つ以上の動作に対応する出力を生成する1つ以上の回路を、ハードウェアが含むことを意味する。コンピュータプログラム、エンジン、またはモジュールが1つ以上の動作を実行するように構成されると言う場合、これは、1つ以上のコンピュータによって実行されるとこの1つ以上のコンピュータにこの1つ以上の動作を実行させる1つ以上のプログラム命令を、コンピュータプログラムが含むことを意味する。
図面に示されて請求項に記載されている動作は特定の順序で示されているが、これらの動作は示されている順序とは異なる順序で実行可能であること、また、一部の動作は省略可能であり、複数回実行可能であり、ならびに/または他の動作と並列におよび/もしくは同時に実行可能であることが理解される。さらに、異なる動作を実行するように構成された異なるシステムコンポーネントの分離は、コンポーネントの分離を要するものと理解されるべきではない。記載されているコンポーネント、モジュール、プログラム、およびエンジンは、1つのシステムとして統合されてもよく、または複数のシステムの一部であってもよい。
特に明記しない限り、上述の代替例は互いに排他的ではなく、特有の利点を達成するためにさまざまな組み合わせで実現され得る。上記特徴のこれらのおよび他の変形例および組み合わせを、請求項によって定義される主題から逸脱することなく利用することができるので、実施例の上述の記載は、請求項によって定義される主題を限定するものとして解釈されるべきではなく、例示するものとして解釈されるべきである。加えて、本明細書に記載されている実施例の提供、ならびに、「などの」および「含む」などと表現されている節は、請求項の主題を特定の実施例に限定するものとして解釈されるべきではなく、むしろ、実施例は、多数の可能な実現例のうちの1つのみを示すことを意図している。さらに、さまざまな図における同一の参照番号は同一または同様の要素を識別し得る。

Claims (20)

  1. 複数の乗算回路を含むスパースシャードを備え、前記スパースシャードは、
    予め定められた最大非ゼロ閾値以下の数の非ゼロ値を含むシャード入力行列を受け、
    複数のベクトル値を含むシャード入力ベクトルを受け、
    前記乗算回路の各々について、前記シャード入力行列のそれぞれの非ゼロ値を受け、
    前記複数の乗算回路によって、ベクトル値に前記シャード入力行列の前記それぞれの非ゼロ値を乗算した1つ以上の積を生成し、
    前記スパースシャードへの出力として、前記1つ以上の積を用いて、前記シャード入力ベクトルを前記シャード入力行列に適用した積であるシャード出力ベクトルを生成するように構成されている、システム。
  2. 前記シャード出力ベクトルの長さは1よりも大きい、請求項1に記載のシステム。
  3. 前記スパースシャードは複数のスパースシャードのうちの1つであり、前記複数のスパースシャードは、
    システム入力行列の部分行列である複数のシャード入力行列を受け、
    システム入力ベクトルの部分ベクトルである複数のシャード入力ベクトルを受け、
    前記複数のスパースシャードによって、前記システム入力ベクトルを前記システム入力行列に適用した積を表すシステム出力ベクトルを生成するように構成されている、請求項1に記載のシステム。
  4. 前記複数のスパースシャードはシストリックアレイとして配列され、前記シストリックアレイは、前記シストリックアレイの列次元に沿ったスパースシャードの1つ以上のグループを含み、
    前記システム出力ベクトルを生成するために、前記1つ以上のプロセッサはさらに、
    前記シストリックアレイの前記列次元に沿ったグループごとに、前記グループ内の各スパースシャードのそれぞれのシャード出力ベクトルを合算してそれぞれの列出力ベクトルを生成し、
    各グループの前記それぞれの列出力ベクトルを連結して前記システム出力ベクトルを生成するように構成されている、請求項3に記載のシステム。
  5. 各乗算回路は、前記スパースシャードについての前記シャード入力行列のそれぞれからの前記それぞれの非ゼロ値を含むそれぞれのレジスタに結合される、請求項1に記載のシステム。
  6. 前記複数の乗算回路における乗算回路の数は、前記予め定められた最大非ゼロ閾値と等しい、請求項5に記載のシステム。
  7. 前記スパースシャードはクロスバー回路をさらに含み、前記スパースシャードはさらに、
    前記クロスバー回路によって、前記シャード入力ベクトルの前記複数のベクトル値を受け、
    前記複数の乗算回路の各々への入力として、前記クロスバー回路によって、前記複数のベクトル値のうちの1つのベクトル値を送るように構成されている、請求項5に記載のシステム。
  8. 前記スパースシャードはさらに、
    前記シャード入力行列の同じ列の非ゼロ値を、前記複数の乗算回路のうちの隣接する乗算回路のレジスタにロードするように構成されている、請求項7に記載のシステム。
  9. 前記スパースシャードはさらに、前記シャード入力行列の各列に沿った非ゼロ値の位置を少なくとも指定する1つ以上の制御値を受けるように構成されており、
    前記スパースシャードの前記クロスバー回路はさらに、
    前記1つ以上の制御値を受け、
    前記1つ以上の制御値に従って、前記シャード入力行列の同じ列に沿った非ゼロ値が乗算されるベクトル値を、隣接する乗算回路に送るように構成されている、請求項8に記載のシステム。
  10. 前記スパースシャードは複数の加算回路をさらに含み、
    前記スパースシャードは1つ以上のセグメントマーカをさらに含み、各セグメントマーカは、前記セグメントマーカにロードされたそれぞれの制御値の値に基づいて、前記複数の加算回路のうちのそれぞれの加算回路への入力をゲート開閉するように構成されており、
    前記スパースシャードはさらに、
    前記1つ以上の制御値の少なくとも一部を前記1つ以上のセグメントマーカにロードするように構成されており、前記シャード入力行列の1列目の非ゼロ値についての加算回路は、前記1列目とは異なる前記シャード入力行列の2列目の非ゼロ値を含む隣接する加算回路の入力を受けることからゲート開閉されており、前記スパースシャードはさらに、
    前記複数の加算回路によって、前記1つ以上の積の1つ以上の合計を生成するように構成されており、前記1つ以上の合計の各々は、前記シャード入力行列の列の1つ以上の非ゼロ値に前記シャード入力ベクトルの1つ以上のそれぞれの値を乗算したそれぞれのセグメント化合計である、請求項9に記載のシステム。
  11. 前記複数の加算回路は並列セグメント化合計回路を形成し、1つ以上の前記セグメント化合計の各々は、セグメントマーカによってゲート開閉されていない隣接する加算回路への出力の合計である、請求項10に記載のシステム。
  12. 前記クロスバー回路は第1のクロスバー回路であり、
    前記スパースシャードは第2のクロスバー回路をさらに含み、前記第2のクロスバー回路は、
    1つ以上の前記セグメント化合計を受け、
    前記1つ以上の制御値に従って1つ以上の前記セグメント化合計を配列して、前記スパースシャードついて前記シャード出力ベクトルのそれぞれを生成するように構成されている、請求項10に記載のシステム。
  13. 前記第2のクロスバー回路はベネシュネットワークを形成し、前記シャード入力行列は正方行列である、請求項12に記載のシステム。
  14. 複数の乗算回路を含むスパースシャードが、予め定められた最大非ゼロ閾値以下の数の非ゼロ値を含むシャード入力行列と、複数のベクトル値を含むシャード入力ベクトルとを受けることと、
    前記乗算回路の各々について、前記シャード入力行列の前記非ゼロ値のそれぞれを受けることと、
    前記スパースシャードの前記複数の乗算回路が、それぞれのベクトル値に前記シャード入力行列の前記非ゼロ値のそれぞれを乗算した1つ以上の積を生成することと、
    前記スパースシャードが、前記スパースシャードへの出力として、前記1つ以上の積を用いて、前記シャード入力ベクトルを前記シャード入力行列に適用した積であるシャード出力ベクトルを生成することとを備える、方法。
  15. 前記スパースシャードが前記シャード出力ベクトルを生成することは、1よりも大きい長さの前記シャード出力ベクトルを生成することを含む、請求項14に記載の方法。
  16. 前記スパースシャードは複数のスパースシャードのうちの1つであり、前記複数のスパースシャードは、
    システム入力行列の部分行列である複数のシャード入力行列を受け、
    システム入力ベクトルの部分ベクトルである複数のシャード入力ベクトルを受け、
    前記複数のスパースシャードによって、前記システム入力ベクトルを前記システム入力ベクトルに適用した積を表すシステム出力ベクトルを生成するように構成されている、請求項14に記載の方法。
  17. 前記複数のスパースシャードはシストリックアレイとして配列され、前記シストリックアレイは、前記シストリックアレイの列次元に沿ったスパースシャードの1つ以上のグループを含み、
    前記システム出力ベクトルを生成することは、
    前記シストリックアレイの前記列次元に沿ったグループごとに、前記グループ内の各スパースシャードのそれぞれのシャード出力ベクトルを合算してそれぞれの列出力ベクトルを生成することと、
    各グループの前記それぞれの列出力ベクトルを連結して前記システム出力ベクトルを生成することとを含む、請求項16に記載の方法。
  18. 各乗算回路は、前記スパースシャードについての前記シャード入力行列のそれぞれからのそれぞれの非ゼロ値を含むそれぞれのレジスタに結合される、請求項14に記載の方法。
  19. 前記複数の乗算回路における乗算回路の数は、前記予め定められた最大非ゼロ閾値と等しい、請求項18に記載の方法。
  20. 複数のスパースシャードを含むシステムによって実行されると前記システムに動作を実行させる命令を格納した1つ以上の非一時的なコンピュータ読取可能記憶媒体であって、前記動作は、
    複数の乗算回路を含むスパースシャードによって、予め定められた最大非ゼロ閾値以下の数の非ゼロ値を含むシャード入力行列と、複数のベクトル値を含むシャード入力ベクトルとを受けることと、
    前記乗算回路の各々について、前記シャード入力行列のそれぞれの非ゼロ値を受けることと、
    前記スパースシャードの前記複数の乗算回路によって、それぞれのベクトル値に前記それぞれの非ゼロ値を乗算した1つ以上の積を生成することと、
    前記スパースシャードへの出力として、前記1つ以上の積を用いて、前記シャード入力ベクトルを前記シャード入力行列に適用した積であるシャード出力ベクトルを生成することとを備える、1つ以上の非一時的なコンピュータ読取可能記憶媒体。
JP2023206881A 2021-05-25 2023-12-07 ハードウェアにおけるスパース行列乗算 Pending JP2024028901A (ja)

Applications Claiming Priority (3)

Application Number Priority Date Filing Date Title
US17/329,259 2021-05-25
US17/329,259 US20220382829A1 (en) 2021-05-25 2021-05-25 Sparse matrix multiplication in hardware
JP2021207147A JP7401513B2 (ja) 2021-05-25 2021-12-21 ハードウェアにおけるスパース行列乗算

Related Parent Applications (1)

Application Number Title Priority Date Filing Date
JP2021207147A Division JP7401513B2 (ja) 2021-05-25 2021-12-21 ハードウェアにおけるスパース行列乗算

Publications (1)

Publication Number Publication Date
JP2024028901A true JP2024028901A (ja) 2024-03-05

Family

ID=80222142

Family Applications (2)

Application Number Title Priority Date Filing Date
JP2021207147A Active JP7401513B2 (ja) 2021-05-25 2021-12-21 ハードウェアにおけるスパース行列乗算
JP2023206881A Pending JP2024028901A (ja) 2021-05-25 2023-12-07 ハードウェアにおけるスパース行列乗算

Family Applications Before (1)

Application Number Title Priority Date Filing Date
JP2021207147A Active JP7401513B2 (ja) 2021-05-25 2021-12-21 ハードウェアにおけるスパース行列乗算

Country Status (5)

Country Link
US (1) US20220382829A1 (ja)
EP (1) EP4095719A1 (ja)
JP (2) JP7401513B2 (ja)
KR (2) KR102601034B1 (ja)
CN (1) CN114329329A (ja)

Families Citing this family (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US11940907B2 (en) * 2021-06-25 2024-03-26 Intel Corporation Methods and apparatus for sparse tensor storage for neural network accelerators
WO2024108584A1 (zh) * 2022-11-25 2024-05-30 华为技术有限公司 稀疏算子处理方法及装置

Also Published As

Publication number Publication date
US20220382829A1 (en) 2022-12-01
JP7401513B2 (ja) 2023-12-19
KR102601034B1 (ko) 2023-11-09
KR20220159257A (ko) 2022-12-02
EP4095719A1 (en) 2022-11-30
KR20230155417A (ko) 2023-11-10
CN114329329A (zh) 2022-04-12
JP2022181161A (ja) 2022-12-07

Similar Documents

Publication Publication Date Title
Kung Systolic algorithms for the CMU WARP processor
EP3373210B1 (en) Transposing neural network matrices in hardware
AU2008202591B2 (en) High speed and efficient matrix multiplication hardware module
JP2024028901A (ja) ハードウェアにおけるスパース行列乗算
JP6715900B2 (ja) ニューラルネットワークのパラメータを適応させるための方法および装置
KR102252137B1 (ko) 계산 장치 및 방법
JP2021507335A (ja) 行列入力を行列プロセッサ用のベクトル化入力に変換するためのシステムおよび方法
US20230026006A1 (en) Convolution computation engine, artificial intelligence chip, and data processing method
CN109726822B (zh) 运算方法、装置及相关产品
JP2022541721A (ja) 効率的な乗算のための代替数字形式をサポートするシステムおよび方法
US10884736B1 (en) Method and apparatus for a low energy programmable vector processing unit for neural networks backend processing
JPH07117948B2 (ja) コンピユータ装置
WO2023065701A1 (zh) 内积处理部件、任意精度计算设备、方法及可读存储介质
Arredondo-Velazquez et al. A streaming architecture for Convolutional Neural Networks based on layer operations chaining
TW202020654A (zh) 具有壓縮進位之數位電路
US7653676B2 (en) Efficient mapping of FFT to a reconfigurable parallel and pipeline data flow machine
CN112446007A (zh) 一种矩阵运算方法、运算装置以及处理器
Nash Computationally efficient systolic architecture for computing the discrete Fourier transform
WO2021168644A1 (zh) 数据处理装置、电子设备和数据处理方法
Raha et al. Efficient hardware acceleration of emerging neural networks for embedded machine learning: An industry perspective
Stanković The Reed-Muller-Fourier transform—computing methods and factorizations
CN112889072A (zh) 用于降低功率消耗的系统、方法和装置
Gerdt et al. Some algorithms for calculating unitary matrices for quantum circuits
US6718465B1 (en) Reconfigurable inner product processor architecture implementing square recursive decomposition of partial product matrices
Mozafari et al. Implementing convolutional neural networks using hartley stochastic computing with adaptive rate feature map compression