4 people 914d89fa0b fix tensor cores for gfx1201 (#9838) ...

* fix tensor cores for gfx1201

* fix typo

* fix python wmma

* AMDLLVMRenderer with arch + AMDLLVM tensor_cores

* fix ci

* clean up

* more tensor cores for RDNA4

* fix half/half, bfloat16/float, bfloat16/bfloat16 for amd_llvm

---------

Co-authored-by: nimlgen <[email protected]>
Co-authored-by: George Hotz <[email protected]>

2025-04-24 14:57:41 -04:00

..

codegen

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

engine

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

frontend

add onnx frontend stub [pr] (#9558)

2025-03-24 12:24:34 +08:00

nn

nn.state docs cleanup (#8332)

2025-03-18 17:16:40 -04:00

renderer

fix tensor cores for gfx1201 (#9838)

2025-04-24 14:57:41 -04:00

runtime

fix tensor cores for gfx1201 (#9838)

2025-04-24 14:57:41 -04:00

shape

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

viz

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

__init__.py

UOp.variable (#7010)

2024-10-12 18:20:44 +08:00

device.py

improve bf16 case for is_dtype_supported [pr] (#10034)

2025-04-24 14:03:57 -03:00

dtype.py

FP8s truncate (#9937)

2025-04-22 19:12:49 -04:00

gradient.py

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

helpers.py

[bounty] [pr] index validation with z3 (#9981)

2025-04-24 08:06:08 -04:00

ops.py

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

py.typed

add a single py.typed (#6083)

2024-08-14 17:31:46 -07:00

spec.py

hotfix: handle bad z3 install like z3 import fail

2025-04-24 08:09:40 -04:00

tensor.py

create randperm and support pytorch backend (#10019)

2025-04-24 07:29:02 -04:00

upat.py

don't use ops name, just keep sink (#9922)

2025-04-18 08:59:18 +01:00