Sophia Wisdom 2d9a557427 torch.bfloat16 support in cutlass python (#1037) ...

* torch.bfloat16 support in cutlass python

* Update datatypes.py

2023-08-16 11:38:53 -04:00

..

backend

CUTLASS 3.2 (#1024)

2023-08-07 20:50:32 -04:00

cpp

CUTLASS 3.2 (#1024)

2023-08-07 20:50:32 -04:00

emit

CUTLASS 3.2 (#1024)

2023-08-07 20:50:32 -04:00

op

CUTLASS 3.2 (#1024)

2023-08-07 20:50:32 -04:00

utils

torch.bfloat16 support in cutlass python (#1037)

2023-08-16 11:38:53 -04:00

__init__.py

CUTLASS 3.2 (#1024)

2023-08-07 20:50:32 -04:00

epilogue.py

CUTLASS 3.1 (#915)

2023-04-14 23:19:34 -04:00

library_defaults.py

CUTLASS 3.2 (#1024)

2023-08-07 20:50:32 -04:00

swizzle.py

CUTLASS 3.1 (#915)

2023-04-14 23:19:34 -04:00