Kaixhin/main.py

Last active October 20, 2017 20:19

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/Kaixhin/8af7261dc790e8c899b4c2b2db97186e.js"></script>
Save Kaixhin/8af7261dc790e8c899b4c2b2db97186e to your computer and use it in GitHub Desktop.

Download ZIP

Device-agnostic PyTorch code

Raw

main.py

	import torch


	def cast(cuda):
	if cuda:
	return lambda x: x.cuda()
	else:
	return lambda x: x


	torch.cast = cast(False)
	a = torch.cast(torch.rand(3, 3))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment