fp8-gemm / flake.nix
liangsu9988's picture
Publish PI0.5 Thor kernels from 503100c (torch213-cxx11-cu130-aarch64-linux)
9022070 verified
Raw
History Blame
486 Bytes
{
description = "Flake for FlashRT FP8 GEMM kernels";
inputs = {
# Based on huggingface/kernels@e9152aa with only the stale CUTLASS 4.5.2
# fixed-output hash updated. Return to upstream after the hash fix lands.
kernel-builder.url =
"github:LiangSu8899/kernels/d720fa90fb9cd92d1bc60a9dc5c55bef2aafabb8";
};
outputs =
{
self,
kernel-builder,
}:
kernel-builder.lib.genKernelFlakeOutputs {
inherit self;
path = ./.;
};
}