Skip to content

Fix order of tensor operations in dtw_cpu call - #2825

Open
mishazawa wants to merge 1 commit into
openai:mainfrom
mishazawa:patch-1
Open

Fix order of tensor operations in dtw_cpu call#2825
mishazawa wants to merge 1 commit into
openai:mainfrom
mishazawa:patch-1

Conversation

@mishazawa

Copy link
Copy Markdown

convert to f64 after moving to CPU. This will work on MPS device (apple m1)

original error:

    return dtw_cpu(x.double().cpu().numpy())
                   ~~~~~~~~^^
TypeError: Cannot convert a MPS Tensor to float64 dtype as the MPS framework doesn't support float64. 
Please use float32 instead.

convert to f64 after moving to CPU. This will work on MPS device (apple m1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant