[kernel slimming] Move fast_hadamard_transform to jit_kernel (#18475)
This commit is contained in:
@@ -936,15 +936,6 @@ void es_sm100_mxfp8_blockscaled_grouped_quant(
|
||||
torch::Tensor& quant_output,
|
||||
torch::Tensor& scale_factor);
|
||||
|
||||
/*
|
||||
* From fast-hadamard-transform
|
||||
*/
|
||||
torch::Tensor fast_hadamard_transform(torch::Tensor& x, double scale);
|
||||
torch::Tensor fast_hadamard_transform_12N(torch::Tensor& x, double scale);
|
||||
torch::Tensor fast_hadamard_transform_20N(torch::Tensor& x, double scale);
|
||||
torch::Tensor fast_hadamard_transform_28N(torch::Tensor& x, double scale);
|
||||
torch::Tensor fast_hadamard_transform_40N(torch::Tensor& x, double scale);
|
||||
|
||||
/*
|
||||
* From flashmla
|
||||
*/
|
||||
|
||||
Reference in New Issue
Block a user