cos/exp on AMDGPU.jl

[vchuravy]

These are currently failing due to missing adjoints for the OCL functions

Enzyme.jl/test/amdgpu.jl

Lines 41 to 72 in 0995d7a

	# @testset "exp_kernel" begin
	# A = AMDGPU.ones(64,)
	# @roc groupsize=length(A) exp_kernel(A)
	# A = AMDGPU.ones(64,)
	# dA = similar(A)
	# dA .= 1
	# @roc groupsize=length(A) grad_exp_kernel(A, dA)
	# @test all(dA .== exp(1.f0))
	# end
	function cos_kernel(A)
	i = workitemIdx().x
	if i <= length(A)
	A[i] = cos(A[i])
	end
	return nothing
	end
	function grad_cos_kernel(A, dA)
	autodiff_deferred(Reverse, cos_kernel, Const, Duplicated(A, dA))
	return nothing
	end
	# @testset "cos_kernel" begin
	# A = AMDGPU.ones(64,)
	# @roc groupsize=length(A) cos_kernel(A)
	# A = AMDGPU.ones(64,)
	# dA = similar(A)
	# dA .= 1
	# @roc groupsize=length(A) grad_cos_kernel(A, dA)
	# @test all(dA .≈ -sin(1.f0))
	# end

[wsmoses]

Resolved by #1537