Data processing apparatus and methods tensor transform operation

A computer-implemented method of performing a transform operation on an input tensor by a data processing apparatus for a processing resource, said input tensor being formed of a plurality of blocks, each block being a portion of said input tensor capable of being operated on independently of each other, said method comprising:

• communicating by communication circuitry with a control module and a shared storage of said processing resource;
• performing by processing circuitry said transform operation, said processing circuitry comprising sub-block processing circuitry and transformation circuitry;
• and
• storing, by a local tensor transform unit storage, a transform operation output from said processing circuitry;
• wherein said method further comprises:
• receiving, by said communication circuitry, one or more transform parameters;
• dividing, by said sub-block processing circuitry, a first block of said input tensor into one or more input sub-blocks wherein a size of said one or more input sub-blocks is based on said one or more transform parameters and a maximum size of said input sub-blocks wherein the maximum size corresponds to a maximum amount of data that can be stored in the local tensor transform unit storage, each input sub-block being a portion of said first block of said input tensor corresponding to a processing unit of said processing circuitry and capable of being operated on independently of each other;
• upon dividing said input tensor into one or more input sub-blocks, further dividing by said sub-block processing circuitry a first input sub-block into a plurality of bricks, each brick being a portion of said first input sub-block capable of being operated on independently of each other;
• subsequent to the dividing of said first input sub-block into said plurality of bricks:
• reading by said communication circuitry a first brick of said plurality of bricks from said shared storage;
• and
• performing, by said transformation circuitry, said transform operation on said first input brick based on said one or more transform parameters, wherein operating on each brick of said plurality of bricks in turn generates a first output sub-block corresponding to said first input sub-block;
• upon generating said first output sub-block, storing by said transformation circuitry said first output sub-block to said local tensor transform unit storage;
• and
• writing by said transformation circuitry said first output sub-block to said shared storage.

A program product stored on a non-transitory computer-readable medium computer program comprising instructions for controlling a host data processing apparatus to provide an instruction execution environment to perform a transform operation on an input tensor, said input tensor being formed of a plurality of blocks, each block being a portion of said input tensor capable of being operated on independently of each other, said instruction execution environment comprising:

• communication program logic to communicate with a control module and a shared storage of said host data processing apparatus;
• processing program logic to perform said transform operation, said processing program logic comprising sub-block processing program logic and transformation program logic;
• and
• a local tensor transform unit storage to store a transform operation output from said processing program logic;
• wherein said communication program logic is configured to:
• receive one or more transform parameters;
• read a first input sub-block from said shared storage, said first input sub-block being a portion of a first block of said input tensor corresponding to a processing unit of said processing program logic;
• and
• write a first output sub-block to said shared storage,
• wherein said sub-block processing program logic is configured to:
• divide said first block of said first input tensor into one or more input sub-blocks capable of being operated on independently of each other wherein a size of said input sub-blocks is based on said one or more transform parameters and a maximum size of said input sub-blocks wherein the maximum size corresponds to a maximum amount of data that can be stored in the local tensor transform unit storage;
• and
• upon division of said input tensor into one or more input sub-blocks, further divide, by said sub-block processing circuitry, a first input sub-block into a plurality of bricks, each brick being a portion of said first input sub-block capable of being operated on independently of each other;
• wherein said transformation circuitry is configured to, subsequent to division of said first input sub-block into said plurality of bricks:
• perform said transform operation on a first brick of said plurality of bricks based on said one or more transform parameters, wherein operation on each brick of said plurality of bricks in turn generates a first output sub-block corresponding to said first input sub-block;
• and
• upon generation of said first output sub-block, write said first output sub-block to said local tensor transform unit storage.