// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2023 SiFive
 */

#include <linux/export.h>
#include <linux/preempt.h>

#include <asm/csr.h>
#include <asm/fpu.h>
#include <asm/processor.h>
#include <asm/switch_to.h>

void kernel_fpu_begin(void)
{
	preempt_disable();
	fstate_save(current, task_pt_regs(current));
	csr_set(CSR_SSTATUS, SR_FS);
}
EXPORT_SYMBOL_GPL(kernel_fpu_begin);

void kernel_fpu_end(void)
{
	csr_clear(CSR_SSTATUS, SR_FS);
	fstate_restore(current, task_pt_regs(current));
	preempt_enable();
}
EXPORT_SYMBOL_GPL(kernel_fpu_end);