docs.unity.cn
    Show / Hide Table of Contents

    Class BurstBLAS

    Burst specific BLAS implementation

    Inheritance
    Object
    BurstBLAS
    Inherited Members
    Object.ToString()
    Object.Equals(Object)
    Object.Equals(Object, Object)
    Object.ReferenceEquals(Object, Object)
    Object.GetHashCode()
    Object.GetType()
    Object.MemberwiseClone()
    Namespace: Unity.Barracuda
    Syntax
    [Preserve]
    public class BurstBLAS : BLASPlugin

    Methods

    IsCurrentPlatformSupported()

    Query if current platform is supported by the BLAS plugin

    Declaration
    public bool IsCurrentPlatformSupported()
    Returns
    Type Description
    Boolean

    true if plugin supports current platform

    Implements
    BLASPlugin.IsCurrentPlatformSupported()

    IsNative()

    Query if BLAS implementation is coming from platform's native library

    Declaration
    public bool IsNative()
    Returns
    Type Description
    Boolean

    true if BLAS implementation is coming from platform's native library

    Implements
    BLASPlugin.IsNative()

    ScheduleSGEMM(JobHandle, Single*, Int32, Int32, Single*, Int32, Int32, Single*, Int32, Int32, Int32, Boolean, Boolean)

    Launches matrix multiplication C = A x B + C in async-manner

    Declaration
    public JobHandle ScheduleSGEMM(JobHandle dependsOn, float *Ap, int AM, int AN, float *Bp, int BM, int BN, float *Cp, int CM, int CN, int bs, bool transposeA = false, bool transposeB = false)
    Parameters
    Type Name Description
    JobHandle dependsOn

    input data dependency job handle

    Single* Ap

    pointer to the matrix A

    Int32 AM

    matrix A row count

    Int32 AN

    matrix A column count

    Single* Bp

    pointer to the matrix B

    Int32 BM

    matrix B row count

    Int32 BN

    matrix B column count

    Single* Cp

    pointer to the matrix C

    Int32 CM

    matrix C row count

    Int32 CN

    matrix C column count

    Int32 bs

    inner loop block size (if applicable) bs x bs

    Boolean transposeA

    matrix A data is in transposed layout

    Boolean transposeB

    matrix B data is in transposed layout

    Returns
    Type Description
    JobHandle

    job handle

    Implements
    BLASPlugin.ScheduleSGEMM(JobHandle, Single*, Int32, Int32, Single*, Int32, Int32, Single*, Int32, Int32, Int32, Boolean, Boolean)

    SGEMM(Single*, Int32, Int32, Single*, Int32, Int32, Single*, Int32, Int32, Int32, Boolean, Boolean)

    Perform matrix multiplication C = A x B + C

    Declaration
    public void SGEMM(float *Ap, int AM, int AN, float *Bp, int BM, int BN, float *Cp, int CM, int CN, int bs, bool transposeA = false, bool transposeB = false)
    Parameters
    Type Name Description
    Single* Ap

    pointer to the matrix A

    Int32 AM

    matrix A row count

    Int32 AN

    matrix A column count

    Single* Bp

    pointer to the matrix B

    Int32 BM

    matrix B row count

    Int32 BN

    matrix B column count

    Single* Cp

    pointer to the matrix C

    Int32 CM

    matrix C row count

    Int32 CN

    matrix C column count

    Int32 bs

    inner loop block size (if applicable) bs x bs

    Boolean transposeA

    matrix A data is in transposed layout

    Boolean transposeB

    matrix B data is in transposed layout

    Implements
    BLASPlugin.SGEMM(Single*, Int32, Int32, Single*, Int32, Int32, Single*, Int32, Int32, Int32, Boolean, Boolean)
    Back to top Copyright © 2023 Unity Technologies — Terms of use
    Generated by DocFX
    on Tuesday, July 4, 2023