[WHLSL] Metal code generation takes a long time uniquing UnnamedTypes

• Component: New Bugs
• Status: RESOLVED FIXED

[Sam Weinig]

[WHLSL] Metal code generation takes a long time uniquing UnnamedTypes

[Sam Weinig]

Created attachment 375728 [details]
Patch

[Sam Weinig]

This is about a ~1ms improvement (10%) on the metal codegen phase.

[Saam Barati]

Comment on attachment 375728 [details]
Patch

View in context: https://bugs.webkit.org/attachment.cgi?id=375728&action=review

Nice. r=me

> Source/WebCore/ChangeLog:13
> +        the UnnamedType subtree, we should probably do it for the entire AST in a future
> +        change.

Yeah I think this will be a huge speedup. Walking the entire AST on compute_boids is like ~1ms. I think it will be much faster once we remove virtual function calls for AST node type checks.

> Source/WebCore/Modules/webgpu/WHLSL/WHLSLPrepare.cpp:66
> +static constexpr bool dumpPhaseTimes = true;

please revert

> Source/WebCore/Modules/webgpu/WHLSL/AST/WHLSLUnnamedType.h:70
> +    Kind kind() const { return m_kind; }
> +    bool isTypeReference() const { return m_kind == Kind::TypeReference; }
> +    bool isPointerType() const { return m_kind == Kind::PointerType; }
> +    bool isArrayReferenceType() const { return m_kind == Kind::ArrayReferenceType; }
> +    bool isArrayType() const { return m_kind == Kind::ArrayType; }
> +    bool isReferenceType() const { return isPointerType() || isArrayReferenceType(); }

nice

> Source/WebCore/Modules/webgpu/WHLSL/AST/WHLSLUnnamedType.h:76
> +    unsigned hash() const;
> +    bool operator==(const UnnamedType&) const;

It is worth defining these in an inline file somewhere since they're super small functions? Maybe we could do this in a follow up

[Saam Barati]

Comment on attachment 375728 [details]
Patch

View in context: https://bugs.webkit.org/attachment.cgi?id=375728&action=review

>> Source/WebCore/ChangeLog:13
>> +        change.
> 
> Yeah I think this will be a huge speedup. Walking the entire AST on compute_boids is like ~1ms. I think it will be much faster once we remove virtual function calls for AST node type checks.

We should also consider uniquing types right after name resolution runs. Then we make type checking faster too.

[Sam Weinig]

Committed r248395: <https://trac.webkit.org/changeset/248395>

[Radar WebKit Bug Importer]

<rdar://problem/54056881>