- @stub.function()
+ @stub.function(gpu="T4")
def f(i):
return i * i
Last active
November 19, 2023 20:46
-
-
Save fikisipi/3f01cb28c6b5bec204764b3ef5ea8ba8 to your computer and use it in GitHub Desktop.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment