relu / flake.nix

Optimized relu: cpu/cuda/xpu, 1.2-1.85x faster on RTX 4090, benchmarked vs upstream and torch.relu

e873e70 verified about 1 month ago

281 Bytes

	{
	description = "Flake for ReLU kernel";

	inputs = {
	kernel-builder.url = "github:huggingface/kernels";
	};

	outputs =
	{
	self,
	kernel-builder,
	...
	}:
	kernel-builder.lib.genKernelFlakeOutputs {
	inherit self;
	path = ./.;
	};
	}